fix parse_args

2023-04-16 10:35:51 +08:00
parent 0d89c37ef1
commit 0baea45210
3 changed files with 5 additions and 3 deletions
--- a/lama_cleaner/parse_args.py
+++ b/lama_cleaner/parse_args.py
@@ -85,12 +85,13 @@ def parse_args():
    parser.add_argument(
        "--interactive-seg-model",
        default="vit_l",
-        help=AVAILABLE_INTERACTIVE_SEG_MODELS,
+        choices=AVAILABLE_INTERACTIVE_SEG_MODELS,
+        help=INTERACTIVE_SEG_MODEL_HELP
    )
    parser.add_argument(
        "--interactive-seg-device",
        default="cpu",
-        help=AVAILABLE_INTERACTIVE_SEG_DEVICES,
+        choices=AVAILABLE_INTERACTIVE_SEG_DEVICES,
    )
    parser.add_argument(
        "--enable-remove-bg",