From d1dbd0adb106609567ed18ee529cd0306ef1d558 Mon Sep 17 00:00:00 2001
From: Sakura-Luna <53183413+Sakura-Luna@users.noreply.github.com>
Date: Sun, 9 Apr 2023 13:18:53 +0800
Subject: [PATCH] Preview fix

---
 modules/sd_samplers_kdiffusion.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py
index e9f08518f..81522b13c 100644
--- a/modules/sd_samplers_kdiffusion.py
+++ b/modules/sd_samplers_kdiffusion.py
@@ -152,7 +152,9 @@ class CFGDenoiser(torch.nn.Module):
         devices.test_for_nans(x_out, "unet")
 
         if opts.live_preview_content == "Prompt":
-            sd_samplers_common.store_latent(x_out[0:uncond.shape[0]])
+            p_step = len(x_out) // batch_size - 1
+            p_step = p_step if p_step > 1 else 1
+            sd_samplers_common.store_latent(x_out[0:-uncond.shape[0]:p_step])
         elif opts.live_preview_content == "Negative prompt":
             sd_samplers_common.store_latent(x_out[-uncond.shape[0]:])