add --realesrgan-no-half

2023-04-03 13:32:04 +08:00
parent dd1d45aa79
commit 03206fb8d6
3 changed files with 10 additions and 3 deletions
--- a/lama_cleaner/parse_args.py
+++ b/lama_cleaner/parse_args.py
@@ -104,6 +104,11 @@ def parse_args():
        type=str,
        choices=RealESRGANModelNameList,
    )
+    parser.add_argument(
+        "--realesrgan-no-half",
+        action="store_true",
+        help="Disable half precision for RealESRGAN",
+    )
    parser.add_argument("--enable-gfpgan", action="store_true", help=GFPGAN_HELP)
    parser.add_argument(
        "--gfpgan-device", default="cpu", type=str, choices=GFPGAN_AVAILABLE_DEVICES