wip: controlnet

2023-05-11 21:51:58 +08:00
parent e5ac6a105a
commit 87f54bb87e
10 changed files with 117 additions and 29 deletions
--- a/lama_cleaner/parse_args.py
+++ b/lama_cleaner/parse_args.py
@@ -41,10 +41,7 @@ def parse_args():
    parser.add_argument(
        "--sd-controlnet-method",
        default="control_v11p_sd15_inpaint",
-        choices=[
-            "control_v11p_sd15_canny",
-            "control_v11p_sd15_inpaint",
-        ],
+        choices=SD_CONTROLNET_CHOICES,
    )
    parser.add_argument("--sd-local-model-path", default=None, help=SD_LOCAL_MODEL_HELP)
    parser.add_argument(