🐛 fix mps error msg

lenML · Aug 1, 2024 · 928a9b4 · 928a9b4
1 parent 318d33f
commit 928a9b4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/modules/devices/mac_devices.py b/modules/devices/mac_devices.py
@@ -20,8 +20,8 @@ def check_for_mps() -> bool:
     else:
         try:
             return torch.backends.mps.is_available() and torch.backends.mps.is_built()
-        except:
-            logger.warning("MPS garbage collection failed", exc_info=True)
+        except Exception as e:
+            logger.warning("MPS check failed: %s", exc_info=True)
             return False