kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

add mps to device

Browse Source
This commit is contained in:
Qing 2022-12-04 20:49:52 +08:00
parent 6a0ffdc96e
commit 816f4989f5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lama_cleaner/parse_args.py
View File

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