Merge pull request #846 from pedrodanielsantos/main

Fix "Update face_enhancer.py"
hacksider · Dec 23, 2024 · 3a85884 · 3a85884
2 parents ed7a216 + 7472dfb
commit 3a85884
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/modules/processors/frame/face_enhancer.py b/modules/processors/frame/face_enhancer.py
@@ -54,13 +54,17 @@ def get_face_enhancer() -> Any:
     with THREAD_LOCK:
         if FACE_ENHANCER is None:
             model_path = os.path.join(models_dir, "GFPGANv1.4.pth")
-        elif platform.system() == "Darwin":  # Mac OS
-                mps_device = None
-        elif 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]
-        elif:  # Other OS
-                FACE_ENHANCER = gfpgan.GFPGANer(model_path=model_path, upscale=1)  # type: ignore[attr-defined]
+
+            match platform.system():
+                case "Darwin":  # Mac OS
+                    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]
+                    else:
+                        FACE_ENHANCER = gfpgan.GFPGANer(model_path=model_path, upscale=1)  # type: ignore[attr-defined]
+                case _:  # Other OS
+                    FACE_ENHANCER = gfpgan.GFPGANer(model_path=model_path, upscale=1)  # type: ignore[attr-defined]
+
     return FACE_ENHANCER