git status

toolkit/stable_diffusion_model.py

@@ -70,6 +70,7 @@ from typing import TYPE_CHECKING
 from toolkit.print import print_acc
 from diffusers import FluxFillPipeline
 from transformers import AutoModel, AutoTokenizer, Gemma2Model, Qwen2Model, LlamaModel
+from toolkit.memory_management import MemoryManager
 
 if TYPE_CHECKING:
     from toolkit.lora_special import LoRASpecialNetwork
@@ -224,6 +225,8 @@ class StableDiffusion:
         # do not resize control images
         self.use_raw_control_images = False
         
+        self.memory_manager = MemoryManager(self)
+        
     # properties for old arch for backwards compatibility
     @property
     def is_xl(self):