Merge 4860ba84fc
into fdbc29c1a9
commit
8b12692cfe
|
@ -63,8 +63,6 @@ def get_face_swapper() -> Any:
|
||||||
with THREAD_LOCK:
|
with THREAD_LOCK:
|
||||||
if FACE_SWAPPER is None:
|
if FACE_SWAPPER is None:
|
||||||
model_name = "inswapper_128.onnx"
|
model_name = "inswapper_128.onnx"
|
||||||
if "CUDAExecutionProvider" in modules.globals.execution_providers:
|
|
||||||
model_name = "inswapper_128_fp16.onnx"
|
|
||||||
model_path = os.path.join(models_dir, model_name)
|
model_path = os.path.join(models_dir, model_name)
|
||||||
FACE_SWAPPER = insightface.model_zoo.get_model(
|
FACE_SWAPPER = insightface.model_zoo.get_model(
|
||||||
model_path, providers=modules.globals.execution_providers
|
model_path, providers=modules.globals.execution_providers
|
||||||
|
|
Loading…
Reference in New Issue