clean pre_check

This commit is contained in:
Somdev Sangwan 2023-06-04 14:13:08 +05:30 committed by GitHub
parent 102f783a2d
commit 8dbbaf0bdb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -97,10 +97,6 @@ def pre_check():
quit(f"CUDNN version {CUDNN_VERSION} is not supported - please upgrade to 8.9.1") quit(f"CUDNN version {CUDNN_VERSION} is not supported - please upgrade to 8.9.1")
if CUDNN_VERSION > 8910: if CUDNN_VERSION > 8910:
quit(f"CUDNN version {CUDNN_VERSION} is not supported - please downgrade to 8.9.1") quit(f"CUDNN version {CUDNN_VERSION} is not supported - please downgrade to 8.9.1")
else:
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): def get_video_frame(video_path, frame_number = 1):