ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 12:25:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

enh: torch mps support

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-01-05 00:27:44 -08:00
parent 614379a427
commit d477f73c7e
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
backend/open_webui/env.py
View file

@ -54,6 +54,9 @@ else:
DEVICE_TYPE = "cpu" DEVICE_TYPE = "cpu"
if torch.backends.mps.is_available() and torch.backends.mps.is_built():
DEVICE_TYPE = "mps"
#################################### ####################################
# LOGGING # LOGGING
#################################### ####################################