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",