diff --git a/comfy/model_management.py b/comfy/model_management.py
index 1889ab0ac..45922faa3 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -27,6 +27,7 @@ import platform
 import weakref
 import gc
 import os
+import comfy.zluda
 
 class VRAMState(Enum):
     DISABLED = 0    #No vram present: no need to move models to vram