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

bruh moment 11

Browse Source
This commit is contained in:
Somdev Sangwan 2023-05-30 08:11:55 +05:30 committed by GitHub
parent ed2bf00d69
commit 68975425ec
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

@ -9,7 +9,7 @@ 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: 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() 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.")