fix gfpgan cpu/mps device

2023-03-28 16:36:41 +08:00
parent 8af5df491e
commit c4968dd0a9
3 changed files with 14 additions and 6 deletions
--- a/lama_cleaner/parse_args.py
+++ b/lama_cleaner/parse_args.py
@@ -107,7 +107,7 @@ def parse_args():
        help="Enable GFPGAN face restore",
    )
    parser.add_argument(
-        "--gfpgan-device", default="cpu", type=str, choices=["cpu", "cuda"]
+        "--gfpgan-device", default="cpu", type=str, choices=["cpu", "cuda", "mps"]
    )
    parser.add_argument(
        "--enable-gif",