From 7e474b6b16a460112cf3abf12d97bb76730b3203 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sun, 10 Aug 2025 01:34:16 +0300 Subject: [PATCH] Fix pusa extension --- nodes.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/nodes.py b/nodes.py index e505996..e862de0 100644 --- a/nodes.py +++ b/nodes.py @@ -1926,10 +1926,11 @@ class WanVideoSampler: all_indices = [] for entry in extra_latents: add_index = entry["index"] - noise[:, add_index] = entry["samples"].squeeze(0).squeeze(1).to(noise) - log.info(f"Adding extra samples to latent index {add_index}") - all_indices.append(add_index) - + num_extra_frames = entry["samples"].shape[2] + noise[:, add_index:add_index+num_extra_frames] = entry["samples"].to(noise) + log.info(f"Adding extra samples to latent indices {add_index} to {add_index+num_extra_frames-1}") + all_indices.extend(range(add_index, add_index+num_extra_frames)) + latent = noise.to(device)