From 242637db5c69f532bb593bf40f7d67a3c0d64363 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Wed, 3 Sep 2025 22:18:37 +0300 Subject: [PATCH] Accept per latent noise multiplier list for Pusa as well --- nodes.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/nodes.py b/nodes.py index 775ef5a..6471d57 100644 --- a/nodes.py +++ b/nodes.py @@ -2346,7 +2346,11 @@ class WanVideoSampler: all_indices = [] noise_multiplier_list = image_embeds.get("pusa_noise_multipliers", None) if noise_multiplier_list is not None: - noise_multipliers = torch.zeros(latent_video_length) + if len(noise_multiplier_list) != latent_video_length: + noise_multipliers = torch.zeros(latent_video_length) + else: + noise_multipliers = torch.tensor(noise_multiplier_list) + log.info(f"Using Pusa noise multipliers: {noise_multipliers}") if extra_latents is not None and transformer.multitalk_model_type.lower() != "infinitetalk": if noise_multiplier_list is not None: noise_multiplier_list = list(noise_multiplier_list) + [1.0] * (len(all_indices) - len(noise_multiplier_list)) @@ -2361,7 +2365,7 @@ class WanVideoSampler: 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)) - if noise_multipliers is not None: + if noise_multipliers is not None and len(noise_multiplier_list) != latent_video_length: for i, idx in enumerate(all_indices): noise_multipliers[idx] = noise_multiplier_list[i] log.info(f"Using Pusa noise multipliers: {noise_multipliers}")