Fixed CUDA-bound error where GPU-enabled run would result in incorrect model loading

pull/1458/head
nopgadget 2025-08-13 18:32:20 -06:00
parent fdbc29c1a9
commit 03576d503a
1 changed files with 1 additions and 1 deletions

View File

@ -64,7 +64,7 @@ def get_face_swapper() -> Any:
if FACE_SWAPPER is None:
model_name = "inswapper_128.onnx"
if "CUDAExecutionProvider" in modules.globals.execution_providers:
model_name = "inswapper_128_fp16.onnx"
model_name = "inswapper_128.onnx"
model_path = os.path.join(models_dir, model_name)
FACE_SWAPPER = insightface.model_zoo.get_model(
model_path, providers=modules.globals.execution_providers