Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 09:38:23 +02:00
Code Issues Packages Projects Releases Wiki Activity

Preview fix

Browse Source
This commit is contained in:
Sakura-Luna
2023-04-09 13:18:53 +08:00
committed by GitHub
parent 22bcc7be42
commit d1dbd0adb1
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_samplers_kdiffusion.py
+3 -1
View File
@@ -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]:])