From b52e5b92bdc218ecb6e959a5d568eb46c0bae41d Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Wed, 26 Mar 2025 00:14:59 +0200 Subject: [PATCH] Update nodes.py --- nodes.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/nodes.py b/nodes.py index 6b36240..df29450 100644 --- a/nodes.py +++ b/nodes.py @@ -1008,7 +1008,6 @@ class WanVideoTextEncode: encoder.model.to(offload_device) mm.soft_empty_cache() - prompt_embeds_dict = { "prompt_embeds": context, "negative_prompt_embeds": context_null, @@ -1995,7 +1994,9 @@ class WanVideoSampler: if experimental_args is not None: video_attention_split_steps = experimental_args.get("video_attention_split_steps", []) if video_attention_split_steps: - transformer.video_attention_split_steps = [int(x.strip()) for x in video_attention_split_steps.split(",")] + transformer.video_attention_split_steps = [int(x.strip()) for x in video_attention_split_steps.split(",")] + else: + transformer.video_attention_split_steps = [] use_zero_init = experimental_args.get("use_zero_init", True) use_cfg_zero_star = experimental_args.get("cfg_zero_star", False) zero_star_steps = experimental_args.get("zero_star_steps", 0) @@ -2055,7 +2056,7 @@ class WanVideoSampler: noise_pred_uncond=noise_pred_uncond[0].to(intermediate_device) noise_pred_text = noise_pred_cond - + #https://github.com/WeichenFan/CFG-Zero-star/ if use_cfg_zero_star: positive_flat = noise_pred_text.view(batch_size, -1)