mirrors/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update onnxruntime

Browse Source
This commit is contained in:
henryruhs 2024-11-02 11:17:39 +01:00
parent d149a71a1b
commit 871f15fe20
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
facefusion/installer.py
View File

@ -13,10 +13,10 @@ from facefusion.common_helper import is_linux, is_macos, is_windows
ONNXRUNTIMES : Dict[str, Tuple[str, str]] = {}
if is_macos():
ONNXRUNTIMES['default'] = ('onnxruntime', '1.19.2')
ONNXRUNTIMES['default'] = ('onnxruntime', '1.20.0')
else:
ONNXRUNTIMES['default'] = ('onnxruntime', '1.19.2')
ONNXRUNTIMES['cuda'] = ('onnxruntime-gpu', '1.19.2')
ONNXRUNTIMES['default'] = ('onnxruntime', '1.20.0')
ONNXRUNTIMES['cuda'] = ('onnxruntime-gpu', '1.20.0')
ONNXRUNTIMES['openvino'] = ('onnxruntime-openvino', '1.19.0')
if is_linux():
ONNXRUNTIMES['rocm'] = ('onnxruntime-rocm', '1.18.0')

2
requirements.txt
View File

@ -3,7 +3,7 @@ gradio==5.4.0
gradio-rangeslider==0.0.8
numpy==2.1.2
onnx==1.16.1
onnxruntime==1.19.2
onnxruntime==1.20.0
opencv-python==4.10.0.84
psutil==6.1.0
tqdm==4.66.6