Fix model_management import so it doesn't get executed twice.

2023-04-15 18:55:17 -04:00
parent 81d1f00df3
commit 73c3e11e83
7 changed files with 14 additions and 14 deletions
--- a/comfy/ldm/modules/diffusionmodules/model.py
+++ b/comfy/ldm/modules/diffusionmodules/model.py
@@ -7,7 +7,7 @@ from einops import rearrange
 from typing import Optional, Any

 from ldm.modules.attention import MemoryEfficientCrossAttention
-import model_management
+from comfy import model_management

 if model_management.xformers_enabled_vae():
    import xformers