mirror of
https://github.com/facefusion/facefusion.git
synced 2026-06-13 07:47:48 +02:00
Set inswapper according to execution provider
This commit is contained in:
@@ -9,8 +9,7 @@ from onnx import numpy_helper
|
||||
import facefusion.globals
|
||||
import facefusion.processors.frame.core as frame_processors
|
||||
from facefusion import config, process_manager, logger, wording
|
||||
from facefusion.common_helper import is_macos
|
||||
from facefusion.execution import apply_execution_provider_options
|
||||
from facefusion.execution import has_execution_provider, apply_execution_provider_options
|
||||
from facefusion.face_analyser import get_one_face, get_average_face, get_many_faces, find_similar_faces, clear_face_analyser
|
||||
from facefusion.face_masker import create_static_box_mask, create_occlusion_mask, create_region_mask, clear_face_occluder, clear_face_parser
|
||||
from facefusion.face_helper import warp_face_by_face_landmark_5, paste_back
|
||||
@@ -150,7 +149,7 @@ def set_options(key : Literal['model'], value : Any) -> None:
|
||||
|
||||
|
||||
def register_args(program : ArgumentParser) -> None:
|
||||
if is_macos():
|
||||
if has_execution_provider('CoreMLExecutionProvider') or has_execution_provider('OpenVINOExecutionProvider'):
|
||||
face_swapper_model_fallback = 'inswapper_128'
|
||||
else:
|
||||
face_swapper_model_fallback = 'inswapper_128_fp16'
|
||||
|
||||
Reference in New Issue
Block a user