add --no-half arg

2023-01-03 21:30:33 +08:00
parent 6cfc7c30f1
commit 59ee89bd34
5 changed files with 6 additions and 3 deletions
--- a/lama_cleaner/parse_args.py
+++ b/lama_cleaner/parse_args.py
@@ -12,6 +12,7 @@ def parse_args():
        default="lama",
        choices=["lama", "ldm", "zits", "mat", "fcf", "sd1.5", "cv2", "manga", "sd2", "paint_by_example"],
    )
+    parser.add_argument("--no-half", action="store_true", help="SD/PaintByExample model no half precision")
    parser.add_argument(
        "--hf_access_token",
        default="",