Timothy Jaeryang Baek 8 달 전
부모
커밋
a4132322d9
1개의 변경된 파일5개의 추가작업 그리고 3개의 파일을 삭제
  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