UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-03-16 10:28:28 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-01-05 00:28:36 -08:00
parent d477f73c7e
commit a4132322d9
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/open_webui/env.py
View File

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