OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-04 03:37:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: torch mps not working

Browse Source 
Co-Authored-By: Rich Tong <1782087+richtong@users.noreply.github.com>
This commit is contained in:
Timothy Jaeryang Baek 2025-01-06 10:08:12 -08:00
parent 15a182c9d6
commit 960683eced
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/open_webui/env.py
View File

@ -54,6 +54,8 @@ else:
DEVICE_TYPE = "cpu" DEVICE_TYPE = "cpu"
try: try:
import torch
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: except Exception: