From ca2812bae09f337378dc1d70714bf7287e27883a Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Fri, 10 Nov 2023 22:05:25 -0500
Subject: [PATCH] Fix RescaleCFG for batch size > 1.

---
 comfy_extras/nodes_model_advanced.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/comfy_extras/nodes_model_advanced.py b/comfy_extras/nodes_model_advanced.py
index 09d2d907..399123ea 100644
--- a/comfy_extras/nodes_model_advanced.py
+++ b/comfy_extras/nodes_model_advanced.py
@@ -140,6 +140,7 @@ class RescaleCFG:
             uncond = args["uncond"]
             cond_scale = args["cond_scale"]
             sigma = args["sigma"]
+            sigma = sigma.view(sigma.shape[:1] + (1,) * (cond.ndim - 1))
             x_orig = args["input"]
 
             #rescale cfg has to be done on v-pred model output