wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-10 17:22:36 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

Stop LTXVImgToVideoInplace from mutating input latents and dropping noise_mask

Browse Source
This commit is contained in:
ozbayb 2026-05-07 17:18:18 -06:00
parent 25757a53c9
commit 3662807a38
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
comfy_extras/nodes_lt.py
View File

@ -106,12 +106,12 @@ class LTXVImgToVideoInplace(io.ComfyNode):
if bypass: if bypass:
return (latent,) return (latent,)
samples = latent["samples"] samples = latent["samples"].clone()
_, height_scale_factor, width_scale_factor = ( _, height_scale_factor, width_scale_factor = (
vae.downscale_index_formula vae.downscale_index_formula
) )
batch, _, latent_frames, latent_height, latent_width = samples.shape _, _, _, latent_height, latent_width = samples.shape
width = latent_width * width_scale_factor width = latent_width * width_scale_factor
height = latent_height * height_scale_factor height = latent_height * height_scale_factor
@ -124,11 +124,7 @@ class LTXVImgToVideoInplace(io.ComfyNode):
samples[:, :, :t.shape[2]] = t samples[:, :, :t.shape[2]] = t
conditioning_latent_frames_mask = torch.ones( conditioning_latent_frames_mask = get_noise_mask(latent)
(batch, 1, latent_frames, 1, 1),
dtype=torch.float32,
device=samples.device,
)
conditioning_latent_frames_mask[:, :, :t.shape[2]] = 1.0 - strength conditioning_latent_frames_mask[:, :, :t.shape[2]] = 1.0 - strength
return io.NodeOutput({"samples": samples, "noise_mask": conditioning_latent_frames_mask}) return io.NodeOutput({"samples": samples, "noise_mask": conditioning_latent_frames_mask})