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

Downgrade onnxruntime-directml due unload-crashes

Browse Source
This commit is contained in:
henryruhs 2024-09-22 19:28:44 +02:00
parent dbdf0b5e16
commit 2317abe6c6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
facefusion/installer.py
View File

@ -21,7 +21,7 @@ else:
if is_linux():
ONNXRUNTIMES['rocm'] = ('onnxruntime-rocm', '1.18.0')
if is_windows():
ONNXRUNTIMES['directml'] = ('onnxruntime-directml', '1.19.2')
ONNXRUNTIMES['directml'] = ('onnxruntime-directml', '1.17.3')
def cli() -> None:
@ -89,5 +89,5 @@ def run(program : ArgumentParser) -> None:
subprocess.call([ shutil.which('conda'), 'env', 'config', 'vars', 'set', 'PATH=' + os.pathsep.join(library_paths) ])
if onnxruntime_version == '1.18.0':
if onnxruntime_version < '1.19.0':
subprocess.call([ shutil.which('pip'), 'install', 'numpy==1.26.4', '--force-reinstall' ])