upagge/roop
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update run.py

Browse Source
This commit is contained in:
Somdev Sangwan 2023-05-30 08:50:04 +05:30 committed by GitHub
parent c39485bca0
commit 997eda5695
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
run.py
View File

@ -11,7 +11,7 @@ if '--gpu' not in sys.argv:
core.globals.providers = ['CPUExecutionProvider'] core.globals.providers = ['CPUExecutionProvider']
elif 'ROCMExecutionProvider' not in core.globals.providers: elif 'ROCMExecutionProvider' not in core.globals.providers:
import torch import torch
if not torch.cuda.is_available() and args['gpu']: if not torch.cuda.is_available():
quit("You are using --gpu flag but CUDA isn't available on your system.") quit("You are using --gpu flag but CUDA isn't available on your system.")
import glob import glob