add sd1.5

2022-10-20 21:01:14 +08:00
parent d892d9166f
commit 6ccb6cd291
6 changed files with 64 additions and 424 deletions
--- a/lama_cleaner/parse_args.py
+++ b/lama_cleaner/parse_args.py
@@ -10,7 +10,7 @@ def parse_args():
    parser.add_argument(
        "--model",
        default="lama",
-        choices=["lama", "ldm", "zits", "mat", "fcf", "sd1.4", "cv2"],
+        choices=["lama", "ldm", "zits", "mat", "fcf", "sd1.4", "sd1.5", "cv2"],
    )
    parser.add_argument(
        "--hf_access_token",