Skip to content

Commit 5d87985

Browse files
committed
fix a bug in cmd arguments about custom checkpoint and config
1 parent 993feb7 commit 5d87985

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

app_svc.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -429,8 +429,8 @@ def main(args):
429429

430430
if __name__ == "__main__":
431431
parser = argparse.ArgumentParser()
432-
parser.add_argument("--checkpoint-path", type=str, help="Path to the checkpoint file", default=None)
433-
parser.add_argument("--config-path", type=str, help="Path to the config file", default=None)
432+
parser.add_argument("--checkpoint", type=str, help="Path to the checkpoint file", default=None)
433+
parser.add_argument("--config", type=str, help="Path to the config file", default=None)
434434
parser.add_argument("--share", type=str2bool, nargs="?", const=True, default=False, help="Whether to share the app")
435435
parser.add_argument("--fp16", type=str2bool, nargs="?", const=True, help="Whether to use fp16", default=True)
436436
parser.add_argument("--gpu", type=int, help="Which GPU id to use", default=0)

app_vc.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -382,8 +382,8 @@ def main(args):
382382

383383
if __name__ == "__main__":
384384
parser = argparse.ArgumentParser()
385-
parser.add_argument("--checkpoint-path", type=str, help="Path to the checkpoint file", default=None)
386-
parser.add_argument("--config-path", type=str, help="Path to the config file", default=None)
385+
parser.add_argument("--checkpoint", type=str, help="Path to the checkpoint file", default=None)
386+
parser.add_argument("--config", type=str, help="Path to the config file", default=None)
387387
parser.add_argument("--share", type=str2bool, nargs="?", const=True, default=False, help="Whether to share the app")
388388
parser.add_argument("--fp16", type=str2bool, nargs="?", const=True, help="Whether to use fp16", default=True)
389389
parser.add_argument("--gpu", type=int, help="Which GPU id to use", default=0)

0 commit comments

Comments
 (0)