Merge pull request #829 from RishonLi/patch-1

Update face_enhancer.py for apple silicon mps
2024-12-16 22:30:50 +08:00 · 2024-12-16 22:30:50 +08:00 · 5f5fe8890a
parent a9e8f27360 de4f765878
commit 5f5fe8890a
1 changed files with 4 additions and 1 deletions
--- a/modules/processors/frame/face_enhancer.py
+++ b/modules/processors/frame/face_enhancer.py
@ -49,7 +49,10 @@ def get_face_enhancer() -> Any:
    with THREAD_LOCK:
        if FACE_ENHANCER is None:
            model_path = os.path.join(models_dir, 'GFPGANv1.4.pth')
-            FACE_ENHANCER = gfpgan.GFPGANer(model_path=model_path, upscale=1)  # type: ignore[attr-defined]
+            mps_device = None
+            if torch.backends.mps.is_available():
+                mps_device = torch.device("mps")
+            FACE_ENHANCER = gfpgan.GFPGANer(model_path=model_path, upscale=1, device=mps_device)  # type: ignore[attr-defined]
    return FACE_ENHANCER