From 816f4989f5e00ba1285d074d8fe72061518d7ed9 Mon Sep 17 00:00:00 2001 From: Qing Date: Sun, 4 Dec 2022 20:49:52 +0800 Subject: [PATCH] add mps to device --- lama_cleaner/parse_args.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lama_cleaner/parse_args.py b/lama_cleaner/parse_args.py index 6b51e09..7508bd9 100644 --- a/lama_cleaner/parse_args.py +++ b/lama_cleaner/parse_args.py @@ -37,7 +37,7 @@ def parse_args(): action="store_true", help="Enable xFormers optimizations. Requires that xformers package has been installed. See: https://github.com/facebookresearch/xformers" ) - parser.add_argument("--device", default="cuda", type=str, choices=["cuda", "cpu"]) + parser.add_argument("--device", default="cuda", type=str, choices=["cuda", "cpu", "mps"]) parser.add_argument("--gui", action="store_true", help="Launch as desktop app") parser.add_argument( "--gui-size",