Add OpenCV2 model

2022-09-25 21:27:12 +08:00
parent bc98ea256a
commit 35cbbd653c
5 changed files with 88 additions and 3 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"],
+        choices=["lama", "ldm", "zits", "mat", "fcf", "sd1.4", "cv2"],
    )
    parser.add_argument(
        "--hf_access_token",