From 283deb07fa037996c2c5320c1e84449dd8f11798 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 15 Mar 2024 11:07:06 +0100 Subject: [PATCH] Hotfix installer --- facefusion/installer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/facefusion/installer.py b/facefusion/installer.py index b2904b87..56ffbcde 100644 --- a/facefusion/installer.py +++ b/facefusion/installer.py @@ -70,7 +70,7 @@ def run(program : ArgumentParser) -> None: os.remove(wheel_path) else: subprocess.call([ 'pip', 'uninstall', 'onnxruntime', onnxruntime_name, '-y', '-q' ]) - if onnxruntime == 'cuda-12.1': + if onnxruntime == 'cuda-12.2': subprocess.call([ 'pip', 'install', onnxruntime_name + '==' + onnxruntime_version, '--extra-index-url', 'https://pkgs.dev.azure.com/onnxruntime/onnxruntime/_packaging/onnxruntime-cuda-12/pypi/simple', '--force-reinstall' ]) else: subprocess.call([ 'pip', 'install', onnxruntime_name + '==' + onnxruntime_version, '--force-reinstall' ])