Update latents used for preview images in flux

This commit is contained in:
Brandon Rising
2024-09-03 11:48:01 -04:00
committed by Brandon
parent 69f080fb75
commit f3bb592544

View File

@@ -36,7 +36,7 @@ def denoise(
timesteps=t_vec,
guidance=guidance_vec,
)
preview_img = img - t_curr * pred
img = img + (t_prev - t_curr) * pred
if inpaint_extension is not None:
@@ -48,7 +48,7 @@ def denoise(
order=1,
total_steps=len(timesteps),
timestep=int(t_curr),
latents=img,
latents=preview_img,
),
)
step += 1