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-04-19 20:32:15 +02:00
parent 7f4b232a4f
commit 84b9b60e6e
2 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
facefusion/installer.py
View File

@@ -10,15 +10,15 @@ from facefusion.common_helper import is_linux, is_windows
ONNXRUNTIME_SET =\ ONNXRUNTIME_SET =\
{ {
'default': ('onnxruntime', '1.21.0') 'default': ('onnxruntime', '1.21.1')
} }
if is_windows() or is_linux(): if is_windows() or is_linux():
ONNXRUNTIME_SET['cuda'] = ('onnxruntime-gpu', '1.21.0') ONNXRUNTIME_SET['cuda'] = ('onnxruntime-gpu', '1.21.1')
ONNXRUNTIME_SET['openvino'] = ('onnxruntime-openvino', '1.20.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')
if is_linux(): if is_linux():
ONNXRUNTIME_SET['rocm'] = ('onnxruntime-rocm', '1.19.0') ONNXRUNTIME_SET['rocm'] = ('onnxruntime-rocm', '1.21.0')
def cli() -> None: def cli() -> None:
@@ -51,7 +51,7 @@ def run(program : ArgumentParser) -> None:
if python_id in [ 'cp310', 'cp312' ]: if python_id in [ 'cp310', 'cp312' ]:
wheel_name = 'onnxruntime_rocm-' + onnxruntime_version + '-' + python_id + '-' + python_id + '-linux_x86_64.whl' wheel_name = 'onnxruntime_rocm-' + onnxruntime_version + '-' + python_id + '-' + python_id + '-linux_x86_64.whl'
wheel_url = 'https://repo.radeon.com/rocm/manylinux/rocm-rel-6.3.4/' + wheel_name wheel_url = 'https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4/' + wheel_name
subprocess.call([ shutil.which('pip'), 'install', wheel_url, '--force-reinstall' ]) subprocess.call([ shutil.which('pip'), 'install', wheel_url, '--force-reinstall' ])
else: else:
subprocess.call([ shutil.which('pip'), 'install', onnxruntime_name + '==' + onnxruntime_version, '--force-reinstall' ]) subprocess.call([ shutil.which('pip'), 'install', onnxruntime_name + '==' + onnxruntime_version, '--force-reinstall' ])
@@ -86,5 +86,5 @@ def run(program : ArgumentParser) -> None:
subprocess.call([ shutil.which('conda'), 'env', 'config', 'vars', 'set', 'PATH=' + os.pathsep.join(library_paths) ]) subprocess.call([ shutil.which('conda'), 'env', 'config', 'vars', 'set', 'PATH=' + os.pathsep.join(library_paths) ])
if args.onnxruntime in [ 'directml', 'rocm' ]: if args.onnxruntime == 'directml':
subprocess.call([ shutil.which('pip'), 'install', 'numpy==1.26.4', '--force-reinstall' ]) subprocess.call([ shutil.which('pip'), 'install', 'numpy==1.26.4', '--force-reinstall' ])

4
requirements.txt
View File

@@ -1,8 +1,8 @@
gradio==5.25.0 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.0 onnxruntime==1.21.1
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