diff --git a/facefusion/installer.py b/facefusion/installer.py index f4b305d..fbb5f1d 100644 --- a/facefusion/installer.py +++ b/facefusion/installer.py @@ -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.19.0': + if args.onnxruntime in [ 'rocm', 'directml' ]: subprocess.call([ shutil.which('pip'), 'install', 'numpy==1.26.4', '--force-reinstall' ])