Missing imports.

comfyanonymous · Aug 22, 2024 · c0b0da2 · c0b0da2
1 parent c26ca27
commit c0b0da2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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