OSSForks/Deep-Live-Cam
1
0
Fork 0
mirror of https://github.com/hacksider/Deep-Live-Cam.git synced 2025-12-06 00:20:02 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update face_swapper.py

Browse Source 
Optimization based on @SanderGi (experimental) to improve mac FPS
This commit is contained in:
Kenneth Estanislao 2025-10-28 19:16:40 +08:00
parent 3ffa9f38b0
commit b82fdc3f31
1 changed files with 22 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
modules/processors/frame/face_swapper.py
View File

@ -63,14 +63,32 @@ def get_face_swapper() -> Any:
with THREAD_LOCK:
if FACE_SWAPPER is None:
model_path = os.path.join(models_dir, "inswapper_128_fp16.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)
update_status(f"Loading face swapper model from: {model_path}", NAME)
try:
# Ensure the providers list is correctly passed
providers = modules.globals.execution_providers
# print(f"Attempting to load model with providers: {providers}") # Debug print
# Apply CoreML optimization for Mac systems
FACE_SWAPPER = insightface.model_zoo.get_model(
model_path, providers=providers
model_path,
providers=[
(
(
"CoreMLExecutionProvider",
{
"ModelFormat": "MLProgram",
"MLComputeUnits": "CPUAndGPU",
"SpecializationStrategy": "FastPrediction",
"AllowLowPrecisionAccumulationOnGPU": 1,
},
)
if p == "CoreMLExecutionProvider"
else p
)
for p in modules.globals.execution_providers
],
)
update_status("Face swapper model loaded successfully.", NAME)
except Exception as e: