Missing imports.

6 months ago · c0b0da264b
parent c26ca27207
commit c0b0da264b
1 changed files with 2 additions and 0 deletions
--- 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