diff --git a/backend/open_webui/env.py b/backend/open_webui/env.py
index f16f2ea6e..bfe7ced8a 100644
--- a/backend/open_webui/env.py
+++ b/backend/open_webui/env.py
@@ -54,6 +54,8 @@ else:
     DEVICE_TYPE = "cpu"
 
 try:
+    import torch
+
     if torch.backends.mps.is_available() and torch.backends.mps.is_built():
         DEVICE_TYPE = "mps"
 except Exception: