apps/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-10-05 08:17:00 +08:00
Code Issues Actions 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
1 changed files with 2 additions and 3 deletions
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