From f4a7ae6824fed87f0be50906bbc7e2d61d00617b Mon Sep 17 00:00:00 2001 From: henryruhs Date: Thu, 1 Feb 2024 21:46:16 +0100 Subject: [PATCH] Downgrade onnxruntime as of buggy 1.17.0 release --- facefusion/installer.py | 8 ++++---- requirements.txt | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/facefusion/installer.py b/facefusion/installer.py index b5fd1df7..1c54cc97 100644 --- a/facefusion/installer.py +++ b/facefusion/installer.py @@ -16,13 +16,13 @@ TORCH : Dict[str, str] =\ } ONNXRUNTIMES : Dict[str, Tuple[str, str]] =\ { - 'default': ('onnxruntime', '1.17.0') + 'default': ('onnxruntime', '1.16.3') } if platform.system().lower() == 'linux' or platform.system().lower() == 'windows': TORCH['cuda-12.1'] = 'cu121' TORCH['cuda-11.8'] = 'cu118' - ONNXRUNTIMES['cuda-12.1'] = ('onnxruntime-gpu', '1.17.0') - ONNXRUNTIMES['cuda-11.8'] = ('onnxruntime-gpu', '1.17.0') + ONNXRUNTIMES['cuda-12.1'] = ('onnxruntime-gpu', '1.16.3') + ONNXRUNTIMES['cuda-11.8'] = ('onnxruntime-gpu', '1.16.3') ONNXRUNTIMES['openvino'] = ('onnxruntime-openvino', '1.16.0') if platform.system().lower() == 'linux': TORCH['rocm-5.4.2'] = 'rocm5.4.2' @@ -30,7 +30,7 @@ if platform.system().lower() == 'linux': ONNXRUNTIMES['rocm-5.4.2'] = ('onnxruntime-rocm', '1.16.3') ONNXRUNTIMES['rocm-5.6'] = ('onnxruntime-rocm', '1.16.3') if platform.system().lower() == 'windows': - ONNXRUNTIMES['directml'] = ('onnxruntime-directml', '1.17.0') + ONNXRUNTIMES['directml'] = ('onnxruntime-directml', '1.16.3') def cli() -> None: diff --git a/requirements.txt b/requirements.txt index d919a320..3d1c41c7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ filetype==1.2.0 gradio==3.50.2 numpy==1.26.2 onnx==1.15.0 -onnxruntime==1.17.0 +onnxruntime==1.16.3 opencv-python==4.9.0.80 psutil==5.9.6 realesrgan==0.3.0