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

use cpu when set --device=cuda on mac

Browse Source
This commit is contained in:
Qing 2023-05-29 21:40:58 +08:00
parent 6674104742
commit 592ec3485b
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lama_cleaner/parse_args.py
View File

@ -197,6 +197,12 @@ def parse_args():
setattr(args, k, v) setattr(args, k, v)
if args.device == "cuda": if args.device == "cuda":
import platform
if platform.system() == "Darwin":
logger.info("MacOS does not support cuda, use cpu instead")
setattr(args, "device", "cpu")
else:
import torch import torch
if torch.cuda.is_available() is False: if torch.cuda.is_available() is False: