diff --git a/comfy_extras/nodes_model_advanced.py b/comfy_extras/nodes_model_advanced.py
index 09d2d9072b2..399123eaa2e 100644
--- a/comfy_extras/nodes_model_advanced.py
+++ b/comfy_extras/nodes_model_advanced.py
@@ -140,6 +140,7 @@ def rescale_cfg(args):
             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