Timothy Jaeryang Baek 4 hónapja
szülő
commit
a4132322d9
1 módosított fájl, 5 hozzáadás és 3 törlés
  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:
 else:
     DEVICE_TYPE = "cpu"
     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
 # LOGGING