diff --git a/comfy/lora.py b/comfy/lora.py
index 7f867e7..9d8a790 100644
--- a/comfy/lora.py
+++ b/comfy/lora.py
@@ -17,6 +17,8 @@
 """
 
 import comfy.utils
+import comfy.model_management
+import comfy.model_base
 import logging
 import torch