Timothy Jaeryang Baek 4 months ago
parent
commit
a4132322d9
1 changed files with 5 additions and 3 deletions
  1. 5 3
      backend/open_webui/env.py

+ 5 - 3
backend/open_webui/env.py

@@ -53,9 +53,11 @@ if USE_CUDA.lower() == "true":
 else:
     DEVICE_TYPE = "cpu"
 
-
-if torch.backends.mps.is_available() and torch.backends.mps.is_built():
-    DEVICE_TYPE = "mps"
+try:
+    if torch.backends.mps.is_available() and torch.backends.mps.is_built():
+        DEVICE_TYPE = "mps"
+except Exception:
+    pass
 
 ####################################
 # LOGGING