Fix CausalForcingSampler.

This commit is contained in:
Talmaj Marinc 2026-03-20 20:37:49 +01:00
parent c0de57725b
commit 0836390c27

View File

@ -246,11 +246,10 @@ class CausalForcingSampler(io.ComfyNode):
pbar.update(1) pbar.update(1)
current_start_frame += block_frames current_start_frame += block_frames
# Apply latent format scaling # Denormalize latents because VAEDecode expects raw latents.
latent_format = comfy.latent_formats.Wan21() latent_format = comfy.latent_formats.Wan21()
output_scaled = latent_format.process_in(output.float().cpu()) output_denorm = latent_format.process_out(output.float().cpu())
return io.NodeOutput({"samples": output_denorm})
return io.NodeOutput({"samples": output_scaled})
def _lookup_sigma(sigmas, timesteps, t_val): def _lookup_sigma(sigmas, timesteps, t_val):