Next (#122)
* Clear VRAM of face analyser on post process * Mark as NEXT * Reduce tensorflow memory to 512 MB * Cosmetics on installer * Add is_download_done to pre_process() hook to prevent errors * Use latest onnxruntime * Testing for download methods, Make get_download_size more robust * Testing for download methods * Introduce --skip-download argument * Catch exception causes by a firewall * Looks stable to me
This commit is contained in:
@@ -13,11 +13,11 @@ from facefusion import metadata, wording
|
||||
|
||||
ONNXRUNTIMES : Dict[str, Tuple[str, str]] =\
|
||||
{
|
||||
'cpu': ('onnxruntime', '1.15.1'),
|
||||
'cuda': ('onnxruntime-gpu', '1.15.1'),
|
||||
'cpu': ('onnxruntime', '1.16.0 '),
|
||||
'cuda': ('onnxruntime-gpu', '1.16.0'),
|
||||
'coreml-legacy': ('onnxruntime-coreml', '1.13.1'),
|
||||
'coreml-silicon': ('onnxruntime-silicon', '1.14.2'),
|
||||
'directml': ('onnxruntime-directml', '1.15.1'),
|
||||
'directml': ('onnxruntime-directml', '1.16.0'),
|
||||
'openvino': ('onnxruntime-openvino', '1.15.0')
|
||||
}
|
||||
|
||||
@@ -60,6 +60,6 @@ def run() -> None:
|
||||
wheel_name = '-'.join([ 'onnxruntime_silicon', onnxruntime_version, python_id, python_id, 'macosx_12_0_arm64.whl' ])
|
||||
wheel_path = os.path.join(tempfile.gettempdir(), wheel_name)
|
||||
wheel_url = 'https://github.com/cansik/onnxruntime-silicon/releases/download/v' + onnxruntime_version + '/' + wheel_name
|
||||
subprocess.call([ 'curl', wheel_url, '-o', wheel_path, '-L' ])
|
||||
subprocess.call([ 'curl', '--silent', '--location', '--continue-at', '-', '--output', wheel_path, wheel_url ])
|
||||
subprocess.call([ 'pip', 'install', wheel_path ])
|
||||
os.remove(wheel_path)
|
||||
|
||||
Reference in New Issue
Block a user