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

fix providers for non-nvidia

Browse Source
This commit is contained in:
Somdev Sangwan 2023-06-04 14:03:55 +05:30 committed by GitHub
parent 2606be04ba
commit 102f783a2d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
roop/core.py
View File

@ -98,7 +98,9 @@ def pre_check():
if CUDNN_VERSION > 8910:
quit(f"CUDNN version {CUDNN_VERSION} is not supported - please downgrade to 8.9.1")
else:
roop.globals.providers = ['CPUExecutionProvider']
for provider in ['CUDAExecutionProvider', 'TensorrtExecutionProvider']:
if provider in roop.globals.providers:
roop.globals.providers = roop.globals.providers.remove(provider)
def get_video_frame(video_path, frame_number = 1):