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]