From 102f783a2d20c9e3c7545947045d44076d3011bd Mon Sep 17 00:00:00 2001 From: Somdev Sangwan Date: Sun, 4 Jun 2023 14:03:55 +0530 Subject: [PATCH] fix providers for non-nvidia --- roop/core.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/roop/core.py b/roop/core.py index 65c0a2d..1e56943 100755 --- a/roop/core.py +++ b/roop/core.py @@ -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): @@ -250,4 +252,4 @@ def run(): create_test_preview ) - window.mainloop() \ No newline at end of file + window.mainloop()