jones/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
2025-05-18 18:12:56 +02:00
parent 7c2308f19e
commit 94535ab9dc
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
facefusion/installer.py
View File

@@ -10,10 +10,10 @@ from facefusion.common_helper import is_linux, is_windows
ONNXRUNTIME_SET =\ ONNXRUNTIME_SET =\
{ {
'default': ('onnxruntime', '1.21.1') 'default': ('onnxruntime', '1.22.0')
} }
if is_windows() or is_linux(): if is_windows() or is_linux():
ONNXRUNTIME_SET['cuda'] = ('onnxruntime-gpu', '1.21.1') ONNXRUNTIME_SET['cuda'] = ('onnxruntime-gpu', '1.22.0')
ONNXRUNTIME_SET['openvino'] = ('onnxruntime-openvino', '1.21.0') ONNXRUNTIME_SET['openvino'] = ('onnxruntime-openvino', '1.21.0')
if is_windows(): if is_windows():
ONNXRUNTIME_SET['directml'] = ('onnxruntime-directml', '1.17.3') ONNXRUNTIME_SET['directml'] = ('onnxruntime-directml', '1.17.3')

2
requirements.txt
View File

@@ -2,7 +2,7 @@ gradio==5.25.2
gradio-rangeslider==0.0.8 gradio-rangeslider==0.0.8
numpy==2.2.4 numpy==2.2.4
onnx==1.17.0 onnx==1.17.0
onnxruntime==1.21.1 onnxruntime==1.22.0
opencv-python==4.11.0.86 opencv-python==4.11.0.86
psutil==7.0.0 psutil==7.0.0
tqdm==4.67.1 tqdm==4.67.1