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

Numpy fix for older onnxruntime

Browse Source
This commit is contained in:
henryruhs 2024-09-09 14:16:09 +02:00
parent 38dcca4b51
commit 5626810f5c
2 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
facefusion/installer.py
View File

@ -14,15 +14,15 @@ 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.0')
ONNXRUNTIMES['default'] = ('onnxruntime', '1.19.2')
else:
ONNXRUNTIMES['default'] = ('onnxruntime', '1.19.0')
ONNXRUNTIMES['cuda'] = ('onnxruntime-gpu', '1.19.0')
ONNXRUNTIMES['default'] = ('onnxruntime', '1.19.2')
ONNXRUNTIMES['cuda'] = ('onnxruntime-gpu', '1.19.2')
ONNXRUNTIMES['openvino'] = ('onnxruntime-openvino', '1.18.0')
if is_linux():
ONNXRUNTIMES['rocm'] = ('onnxruntime-rocm', '1.18.0')
if is_windows():
ONNXRUNTIMES['directml'] = ('onnxruntime-directml', '1.19.0')
ONNXRUNTIMES['directml'] = ('onnxruntime-directml', '1.19.2')
def cli() -> None:
@ -102,3 +102,6 @@ def run(program : ArgumentParser) -> None:
library_paths = [ library_path for library_path in library_paths if os.path.exists(library_path) ]
subprocess.call([ shutil.which('conda'), 'env', 'config', 'vars', 'set', 'PATH=' + os.pathsep.join(library_paths) ])
if onnxruntime_version == '1.18.0':
subprocess.call([ shutil.which('pip'), 'install', 'numpy==1.26.4', '--force-reinstall' ])

2
requirements.txt
View File

@ -3,7 +3,7 @@ gradio==4.43.0
gradio_rangeslider==0.0.6
numpy==2.1.0
onnx==1.16.2
onnxruntime==1.19.0
onnxruntime==1.19.2
opencv-python==4.10.0.84
psutil==6.0.0
tqdm==4.66.5