From fe25d21fbaf8f08e093ee27d72e20528a53ed070 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 29 Apr 2025 02:07:05 +0300
Subject: [PATCH] Fix audio cfg

silly mistake
---
 nodes.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index af8f9a0..9e239cc 100644
--- a/nodes.py
+++ b/nodes.py
@@ -2898,8 +2898,11 @@ class WanVideoSampler:
                                 **base_params
                             )
                             noise_pred_no_audio = noise_pred_no_audio[0].to(intermediate_device)
-                            noise_pred = noise_pred_uncond + cfg_scale * (noise_pred_no_audio - noise_pred_uncond)
-                            + audio_cfg_scale * (noise_pred_cond - noise_pred_no_audio)
+                            noise_pred = (
+                                noise_pred_uncond
+                                + cfg_scale * (noise_pred_no_audio - noise_pred_uncond)
+                                + audio_cfg_scale * (noise_pred_cond - noise_pred_no_audio)
+                                )
                             return noise_pred, [teacache_state_cond, teacache_state_uncond, teacache_state_audio]
 
                 #batched