Update nodes.py

This commit is contained in:
kijai
2025-07-18 14:41:38 +03:00
parent 8538c517e6
commit 8b8da7d8fe
-3
View File
@@ -2381,11 +2381,8 @@ class WanVideoSampler:
if mask.shape[2] != noise.shape[1]:
mask = torch.cat([torch.zeros(1, noise.shape[0], noise.shape[1] - mask.shape[2], noise.shape[2], noise.shape[3]), mask], dim=2)
if (extra_latents := image_embeds.get("extra_latents", None)) is not None:
if (extra_latents := image_embeds.get("extra_latents", None)) is not None:
encoded_image_latents = extra_latents["samples"].squeeze(0).to(noise)
if (empty_latent_indices := extra_latents.get("empty_latent_indices", None)) is not None and len(empty_latent_indices) > 0:
if (empty_latent_indices := extra_latents.get("empty_latent_indices", None)) is not None and len(empty_latent_indices) > 0:
noise_out = encoded_image_latents.clone()
for idx in empty_latent_indices: