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

Update onnxruntime (second try)

Browse Source
This commit is contained in:
henryruhs 2024-11-03 23:30:45 +01:00
parent 4bffa0d183
commit f89398d686
3 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
facefusion/execution.py
View File

@ -37,7 +37,7 @@ def create_execution_providers(execution_device_id : str, execution_provider_key
execution_providers.append((execution_provider_set.get(execution_provider_key), execution_providers.append((execution_provider_set.get(execution_provider_key),
{ {
'device_id': execution_device_id, 'device_id': execution_device_id,
'cudnn_conv_algo_search': 'EXHAUSTIVE' if use_exhaustive() else 'DEFAULT' 'cudnn_conv_algo_search': 'EXHAUSTIVE' if use_exhaustive() else 'HEURISTIC'
})) }))
if execution_provider_key == 'tensorrt': if execution_provider_key == 'tensorrt':
execution_providers.append((execution_provider_set.get(execution_provider_key), execution_providers.append((execution_provider_set.get(execution_provider_key),

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]] = {} ONNXRUNTIMES : Dict[str, Tuple[str, str]] = {}
if is_macos(): if is_macos():
ONNXRUNTIMES['default'] = ('onnxruntime', '1.19.2') ONNXRUNTIMES['default'] = ('onnxruntime', '1.20.0')
else: else:
ONNXRUNTIMES['default'] = ('onnxruntime', '1.19.2') ONNXRUNTIMES['default'] = ('onnxruntime', '1.20.0')
ONNXRUNTIMES['cuda'] = ('onnxruntime-gpu', '1.19.2') ONNXRUNTIMES['cuda'] = ('onnxruntime-gpu', '1.20.0')
ONNXRUNTIMES['openvino'] = ('onnxruntime-openvino', '1.19.0') ONNXRUNTIMES['openvino'] = ('onnxruntime-openvino', '1.19.0')
if is_linux(): if is_linux():
ONNXRUNTIMES['rocm'] = ('onnxruntime-rocm', '1.18.0') 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 gradio-rangeslider==0.0.8
numpy==2.1.2 numpy==2.1.2
onnx==1.17.0 onnx==1.17.0
onnxruntime==1.19.2 onnxruntime==1.20.0
opencv-python==4.10.0.84 opencv-python==4.10.0.84
psutil==6.1.0 psutil==6.1.0
tqdm==4.66.6 tqdm==4.66.6