From f9179540d2bb510a330667d167ed2cb0d4edcb95 Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Fri, 16 Aug 2024 17:29:42 -0600
Subject: [PATCH] Flush after sampling

---
 toolkit/stable_diffusion_model.py | 2 ++
 1 file changed, 2 insertions(+)

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,