From c0b0da264b30c8583ad54a7646b9d55286d499ca Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Thu, 22 Aug 2024 17:20:39 -0400
Subject: [PATCH] Missing imports.

---
 comfy/lora.py | 2 ++
 1 file changed, 2 insertions(+)

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