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

fix #69

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

5
run.py
View File

@ -9,10 +9,9 @@ if not shutil.which('ffmpeg'):
quit() quit()
if '--gpu' not in sys.argv: if '--gpu' not in sys.argv:
core.globals.providers = ['CPUExecutionProvider'] core.globals.providers = ['CPUExecutionProvider']
else if 'ROCMExecutionProvider' not in core.globals.providers:
if 'ROCMExecutionProvider' not in core.globals.providers:
import torch import torch
if not torch.cuda.is_available(): if not torch.cuda.is_available() and args['gpu']:
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