From d67c96e4b587f0c4cfabb6467f84e5b70eedf701 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 25 Mar 2025 21:30:44 +0200
Subject: [PATCH] Update nodes.py

---
 nodes.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index 48f1c44..6b36240 100644
--- a/nodes.py
+++ b/nodes.py
@@ -2055,6 +2055,8 @@ 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)  
                         negative_flat = noise_pred_uncond.view(batch_size, -1)  
@@ -2062,7 +2064,6 @@ class WanVideoSampler:
                         alpha = optimized_scale(positive_flat,negative_flat)
                         alpha = alpha.view(batch_size, 1, 1, 1)
 
-
                         if (idx <= zero_star_steps) and use_zero_init:
                             noise_pred = noise_pred_text*0.
                         else: