nodes/kijai-ComfyUI-WanVideoWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai
2025-07-17 19:25:03 +03:00
parent d8e90a94c1
commit 8d7504144d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+4 -4
View File
@@ -2342,10 +2342,10 @@ class WanVideoSampler:
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:
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:
print(f"Adding noise to Empty latent index: {idx}")
#print(f"Adding noise to Empty latent index: {idx}")
noise_out[:, idx] = noise[:, idx]
noise = noise_out
else:
@@ -2944,13 +2944,13 @@ class WanVideoSampler:
if scheduler == "flowmatch_pusa":
timestep = timestep.unsqueeze(1).repeat(1, latent_video_length)
if extra_latents is not None:
if empty_latent_indices is not None:
if empty_latent_indices is not None and len(empty_latent_indices) > 0:
# Set timestep to zero for all non-noise (non-empty) indices
non_noise_indices = [i for i in range(timestep.shape[1]) if i not in empty_latent_indices]
timestep[:, non_noise_indices] = 0
else:
timestep[:,0:encoded_image_latents.shape[1]] = 0
print(f"timestep: {timestep}")
#print(f"timestep: {timestep}")
current_step_percentage = idx / len(timesteps)
### latent shift