let5see/IOPaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove cuda-visible-device

Browse Source
This commit is contained in:
Qing
2023-02-07 21:43:37 +08:00
parent 14d9918ef1
commit bb4c44b38d
2 changed files with 0 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lama_cleaner/parse_args.py
View File

@@ -47,7 +47,6 @@ def parse_args():
)
parser.add_argument("--model", default=DEFAULT_MODEL, choices=AVAILABLE_MODELS)
parser.add_argument("--cuda-visible-device", default="")
parser.add_argument("--no-half", action="store_true", help=NO_HALF_HELP)
parser.add_argument("--cpu-offload", action="store_true", help=CPU_OFFLOAD_HELP)
parser.add_argument("--disable-nsfw", action="store_true", help=DISABLE_NSFW_HELP)
@@ -143,14 +142,6 @@ def parse_args():
parser.error(
"torch.cuda.is_available() is False, please use --device cpu or check your pytorch installation"
)
if args.cuda_visible_device:
try:
int(args.cuda_visible_device)
except:
parser.error(
f"invalid --cuda-visible-device: {args.cuda_visible_device}, must be int"
)
os.environ["CUDA_VISIBLE_DEVICES"] = args.cuda_visible_device
if args.device == "mps":
if args.model not in MPS_SUPPORT_MODELS: