Flush after sampling

2024-08-16 17:29:42 -06:00
parent 452e0e286d
commit f9179540d2
1 changed files with 2 additions and 0 deletions
--- 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,