Timothy Jaeryang Baek há 4 meses atrás
pai
commit
a4132322d9
1 ficheiros alterados com 5 adições e 3 exclusões
  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