From 20bedf389ab088dd571a7adbc1269f24f0b89a2e Mon Sep 17 00:00:00 2001
From: Extraltodeus <extraltodeus@gmail.com>
Date: Wed, 27 Mar 2024 05:35:55 +0100
Subject: [PATCH] Update nodes.py

---
 nodes.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/nodes.py b/nodes.py
index 229c0c8..378f827 100644
--- a/nodes.py
+++ b/nodes.py
@@ -87,6 +87,7 @@ class advancedDynamicCFG:
             uncond_pred = args["uncond_denoised"]
             if lerp_uncond:
                 uncond_pred = torch.lerp(cond_pred,uncond_pred,lerp_uncond_strength)
+                uncond_pred = uncond_pred * cond_pred.norm() / uncond_pred.norm()
             cond = input_x - cond_pred
             uncond = input_x - uncond_pred
             sigma = args["sigma"][0]