CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-09-20 22:40:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

Downgrade onnxruntime as of buggy 1.17.0 release

Browse Source
This commit is contained in:
henryruhs
2024-02-01 21:46:16 +01:00
parent 2013653760
commit f4a7ae6824
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/installer.py
+4 -4
View File
@@ -16,13 +16,13 @@ TORCH : Dict[str, str] =\
} }
ONNXRUNTIMES : Dict[str, Tuple[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': if platform.system().lower() == 'linux' or platform.system().lower() == 'windows':
TORCH['cuda-12.1'] = 'cu121' TORCH['cuda-12.1'] = 'cu121'
TORCH['cuda-11.8'] = 'cu118' TORCH['cuda-11.8'] = 'cu118'
ONNXRUNTIMES['cuda-12.1'] = ('onnxruntime-gpu', '1.17.0') ONNXRUNTIMES['cuda-12.1'] = ('onnxruntime-gpu', '1.16.3')
ONNXRUNTIMES['cuda-11.8'] = ('onnxruntime-gpu', '1.17.0') ONNXRUNTIMES['cuda-11.8'] = ('onnxruntime-gpu', '1.16.3')
ONNXRUNTIMES['openvino'] = ('onnxruntime-openvino', '1.16.0') ONNXRUNTIMES['openvino'] = ('onnxruntime-openvino', '1.16.0')
if platform.system().lower() == 'linux': if platform.system().lower() == 'linux':
TORCH['rocm-5.4.2'] = 'rocm5.4.2' 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.4.2'] = ('onnxruntime-rocm', '1.16.3')
ONNXRUNTIMES['rocm-5.6'] = ('onnxruntime-rocm', '1.16.3') ONNXRUNTIMES['rocm-5.6'] = ('onnxruntime-rocm', '1.16.3')
if platform.system().lower() == 'windows': if platform.system().lower() == 'windows':
ONNXRUNTIMES['directml'] = ('onnxruntime-directml', '1.17.0') ONNXRUNTIMES['directml'] = ('onnxruntime-directml', '1.16.3')
def cli() -> None: def cli() -> None:
requirements.txt
+1 -1
View File
@@ -3,7 +3,7 @@ filetype==1.2.0
gradio==3.50.2 gradio==3.50.2
numpy==1.26.2 numpy==1.26.2
onnx==1.15.0 onnx==1.15.0
onnxruntime==1.17.0 onnxruntime==1.16.3
opencv-python==4.9.0.80 opencv-python==4.9.0.80
psutil==5.9.6 psutil==5.9.6
realesrgan==0.3.0 realesrgan==0.3.0