319e3f9652
* Modernize CI * Modernize CI * Modernize CI * Implement dynamic config (#518) * Implement dynamic config * Fix apply config * Move config to general * Move config to general * Move config to general * Add Windows installer * Add --open-browser * Add Windows installer part2 * Use non-commercial license for the installer * Fix create environment in installer * Fix openvino for installer * Fix conda for installer * Fix conda for installer, Remove python and pip as it is part of conda * Improve installer - guess the path * Fix CI * Add missing accept-source-agreements to installer * Install WinGet * Improve WinGet installation steps * Use absolute path for winget * More installer polishing * Add final page to installer, disable version check for Gradio * Remove finish page again * Use NEXT for metadata * Support for /S mode * Use winget-less approach * Improve Conda uninstall * Improve code using platform helpers (#529) * Update dependencies * Feat/fix windows unicode paths (#531) * Fix the Windows unicode path dilemma * Update dependencies * Fix the Windows unicode path dilemma part2 * Remove conda environment on uninstall * Fix uninstall command * Install apps for local user only * Add ultra sharp * Add clear reality * Update README and FUNDING * Update FUNDING.yml * Prevent preview of large videos in Gradio (#540) * Fix order * Refactor temporary file management, Use temporary file for image processing (#542) * Allow webm on target component * Reduce mosaic effect for frame processors * clear static faces on trim frame changes * Fix trim frame component * Downgrade openvino dependency * Prepare next release * Move get_short_path to filesystem, Add/Improve some testing * Prepare installer, Prevent infinite loop for sanitize_path_for_windows * Introduce execution device id * Introduce execution device id * Seems like device id can be a string * Seems like device id can be a string * Make Intel Arc work with OpenVINOExecution * Use latest Git * Update wording * Fix create_float_range * Update preview * Fix Git link
78 lines
3.4 KiB
Python
78 lines
3.4 KiB
Python
from typing import Dict, Tuple
|
|
import sys
|
|
import os
|
|
import tempfile
|
|
import subprocess
|
|
import inquirer
|
|
from argparse import ArgumentParser, HelpFormatter
|
|
|
|
from facefusion import metadata, wording
|
|
from facefusion.common_helper import is_linux, is_macos, is_windows
|
|
|
|
if is_macos():
|
|
os.environ['SYSTEM_VERSION_COMPAT'] = '0'
|
|
|
|
ONNXRUNTIMES : Dict[str, Tuple[str, str]] = {}
|
|
|
|
if is_macos():
|
|
ONNXRUNTIMES['default'] = ('onnxruntime', '1.17.3')
|
|
else:
|
|
ONNXRUNTIMES['default'] = ('onnxruntime', '1.17.3')
|
|
ONNXRUNTIMES['cuda-12.2'] = ('onnxruntime-gpu', '1.17.1')
|
|
ONNXRUNTIMES['cuda-11.8'] = ('onnxruntime-gpu', '1.17.1')
|
|
ONNXRUNTIMES['openvino'] = ('onnxruntime-openvino', '1.15.0')
|
|
if is_linux():
|
|
ONNXRUNTIMES['rocm-5.4.2'] = ('onnxruntime-rocm', '1.16.3')
|
|
ONNXRUNTIMES['rocm-5.6'] = ('onnxruntime-rocm', '1.16.3')
|
|
if is_windows():
|
|
ONNXRUNTIMES['directml'] = ('onnxruntime-directml', '1.17.3')
|
|
|
|
|
|
def cli() -> None:
|
|
program = ArgumentParser(formatter_class = lambda prog: HelpFormatter(prog, max_help_position = 200))
|
|
program.add_argument('--onnxruntime', help = wording.get('help.install_dependency').format(dependency = 'onnxruntime'), choices = ONNXRUNTIMES.keys())
|
|
program.add_argument('--skip-conda', help = wording.get('help.skip_conda'), action = 'store_true')
|
|
program.add_argument('-v', '--version', version = metadata.get('name') + ' ' + metadata.get('version'), action = 'version')
|
|
run(program)
|
|
|
|
|
|
def run(program : ArgumentParser) -> None:
|
|
args = program.parse_args()
|
|
python_id = 'cp' + str(sys.version_info.major) + str(sys.version_info.minor)
|
|
|
|
if not args.skip_conda and 'CONDA_PREFIX' not in os.environ:
|
|
sys.stdout.write(wording.get('conda_not_activated') + os.linesep)
|
|
sys.exit(1)
|
|
if args.onnxruntime:
|
|
answers =\
|
|
{
|
|
'onnxruntime': args.onnxruntime
|
|
}
|
|
else:
|
|
answers = inquirer.prompt(
|
|
[
|
|
inquirer.List('onnxruntime', message = wording.get('help.install_dependency').format(dependency = 'onnxruntime'), choices = list(ONNXRUNTIMES.keys()))
|
|
])
|
|
if answers:
|
|
onnxruntime = answers['onnxruntime']
|
|
onnxruntime_name, onnxruntime_version = ONNXRUNTIMES[onnxruntime]
|
|
|
|
subprocess.call([ 'pip', 'install', '-r', 'requirements.txt', '--force-reinstall' ])
|
|
if onnxruntime == 'rocm-5.4.2' or onnxruntime == 'rocm-5.6':
|
|
if python_id in [ 'cp39', 'cp310', 'cp311' ]:
|
|
rocm_version = onnxruntime.replace('-', '')
|
|
rocm_version = rocm_version.replace('.', '')
|
|
wheel_name = 'onnxruntime_training-' + onnxruntime_version + '+' + rocm_version + '-' + python_id + '-' + python_id + '-manylinux_2_17_x86_64.manylinux2014_x86_64.whl'
|
|
wheel_path = os.path.join(tempfile.gettempdir(), wheel_name)
|
|
wheel_url = 'https://download.onnxruntime.ai/' + wheel_name
|
|
subprocess.call([ 'curl', '--silent', '--location', '--continue-at', '-', '--output', wheel_path, wheel_url ])
|
|
subprocess.call([ 'pip', 'uninstall', wheel_path, '-y', '-q' ])
|
|
subprocess.call([ 'pip', 'install', wheel_path, '--force-reinstall' ])
|
|
os.remove(wheel_path)
|
|
else:
|
|
subprocess.call([ 'pip', 'uninstall', 'onnxruntime', onnxruntime_name, '-y', '-q' ])
|
|
if onnxruntime == 'cuda-12.2':
|
|
subprocess.call([ 'pip', 'install', onnxruntime_name + '==' + onnxruntime_version, '--extra-index-url', 'https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple', '--force-reinstall' ])
|
|
else:
|
|
subprocess.call([ 'pip', 'install', onnxruntime_name + '==' + onnxruntime_version, '--force-reinstall' ])
|