diff --git a/comfy/model_management.py b/comfy/model_management.py
index 35c3bd787..31511f07d 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -25,7 +25,7 @@ import sys
 import platform
 import weakref
 import gc
-import zluda
+import comfy.zluda
 
 class VRAMState(Enum):
     DISABLED = 0    #No vram present: no need to move models to vram