mirrors/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions 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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
facefusion/installer.py
View File

@ -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:

2
requirements.txt
View File

@ -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