mirror of
https://github.com/facefusion/facefusion.git
synced 2026-09-15 20:15:28 +02:00
Henry Ruhs
6f5e4b9d91
* fix image to image for coreml * fix image to image for coreml * Update metadata.py * modernize onnxruntime * introduce cuda@12 and cuda@13 for installer * introduce cuda@12 and cuda@13 for installer * temporary fix for inference pool disaster * temporary fix for inference pool disaster * use onnxruntime 1.24.4 for cuda@12 * fix test * fix test * fix test * add test for job list * switch to has_arena_leak guard * remove directml hack for inference pool * uninstall every onnxruntime first * uninstall every onnxruntime first * restore the directml fix * restore the directml fix * fix directml again * simplify condition
77 lines
2.6 KiB
Python
77 lines
2.6 KiB
Python
import os
|
|
import shutil
|
|
import signal
|
|
import subprocess
|
|
import sys
|
|
from argparse import ArgumentParser, HelpFormatter
|
|
from functools import partial
|
|
from types import FrameType
|
|
|
|
from facefusion import metadata
|
|
from facefusion.common_helper import is_linux, is_windows
|
|
|
|
LOCALES =\
|
|
{
|
|
'install_dependency': 'install the {dependency} package',
|
|
'force_reinstall': 'force reinstall of packages',
|
|
'skip_conda': 'skip the conda environment check',
|
|
'conda_not_activated': 'conda is not activated'
|
|
}
|
|
ONNXRUNTIME_SET =\
|
|
{
|
|
'default': ('onnxruntime', '1.28.0')
|
|
}
|
|
if is_windows() or is_linux():
|
|
ONNXRUNTIME_SET['cuda@12'] = ('onnxruntime-gpu', '1.24.4')
|
|
ONNXRUNTIME_SET['cuda@13'] = ('onnxruntime-gpu', '1.28.0')
|
|
ONNXRUNTIME_SET['openvino'] = ('onnxruntime-openvino', '1.24.1')
|
|
if is_windows():
|
|
ONNXRUNTIME_SET['directml'] = ('onnxruntime-directml', '1.24.4')
|
|
ONNXRUNTIME_SET['qnn'] = ('onnxruntime-qnn', '2.4.0')
|
|
if is_linux():
|
|
ONNXRUNTIME_SET['migraphx'] = ('onnxruntime-migraphx', '1.27.1')
|
|
ONNXRUNTIME_SET['rocm'] = ('onnxruntime-rocm', '1.22.2.post3')
|
|
|
|
|
|
def cli() -> None:
|
|
signal.signal(signal.SIGINT, signal_exit)
|
|
program = ArgumentParser(formatter_class = partial(HelpFormatter, max_help_position = 50))
|
|
program.add_argument('onnxruntime', help = LOCALES.get('install_dependency').format(dependency = 'onnxruntime'), choices = ONNXRUNTIME_SET.keys())
|
|
program.add_argument('--force-reinstall', help = LOCALES.get('force_reinstall'), action = 'store_true')
|
|
program.add_argument('--skip-conda', help = LOCALES.get('skip_conda'), action = 'store_true')
|
|
program.add_argument('-v', '--version', version = metadata.get('name') + ' ' + metadata.get('version'), action = 'version')
|
|
run(program)
|
|
|
|
|
|
def signal_exit(signum : int, frame : FrameType) -> None:
|
|
sys.exit(0)
|
|
|
|
|
|
def run(program : ArgumentParser) -> None:
|
|
args = program.parse_args()
|
|
has_conda = 'CONDA_PREFIX' in os.environ
|
|
|
|
if not args.skip_conda and not has_conda:
|
|
sys.stdout.write(LOCALES.get('conda_not_activated') + os.linesep)
|
|
sys.exit(1)
|
|
|
|
for onnxruntime_package, _ in ONNXRUNTIME_SET.values():
|
|
subprocess.call([ shutil.which('pip'), 'uninstall', onnxruntime_package, '-y', '-q' ], stderr = subprocess.DEVNULL)
|
|
|
|
commands = [ shutil.which('pip'), 'install' ]
|
|
|
|
if args.force_reinstall:
|
|
commands.append('--force-reinstall')
|
|
|
|
with open('requirements.txt') as file:
|
|
|
|
for line in file.readlines():
|
|
__line__ = line.strip()
|
|
|
|
if not __line__.startswith('onnxruntime'):
|
|
commands.append(__line__)
|
|
|
|
onnxruntime_name, onnxruntime_version = ONNXRUNTIME_SET.get(args.onnxruntime)
|
|
commands.append(onnxruntime_name + '==' + onnxruntime_version)
|
|
subprocess.call(commands)
|