add Manga Model

2022-11-18 21:40:12 +08:00
parent d7e2148ce1
commit 08c295a70d
7 changed files with 150 additions and 5 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.5", "cv2"],
+        choices=["lama", "ldm", "zits", "mat", "fcf", "sd1.5", "cv2", "manga"],
    )
    parser.add_argument(
        "--hf_access_token",