diff --git a/toolkit/stable_diffusion_model.py b/toolkit/stable_diffusion_model.py
index 4274ffe4..47ac104b 100644
--- a/toolkit/stable_diffusion_model.py
+++ b/toolkit/stable_diffusion_model.py
@@ -1289,6 +1289,8 @@ class StableDiffusion:
             else:
                 self.assistant_lora.is_active = True
 
+        flush()
+
     def get_latent_noise(
             self,
             height=None,