diff --git a/run.py b/run.py index cf05641..6b46527 100755 --- a/run.py +++ b/run.py @@ -9,7 +9,7 @@ if not shutil.which('ffmpeg'): quit() if '--gpu' not in sys.argv: core.globals.providers = ['CPUExecutionProvider'] -else if 'ROCMExecutionProvider' not in core.globals.providers: +elif 'ROCMExecutionProvider' not in core.globals.providers: import torch if not torch.cuda.is_available() and args['gpu']: quit("You are using --gpu flag but CUDA isn't available on your system.")