Skip to content

Commit

Permalink
Fix RescaleCFG for batch size > 1.
Browse files Browse the repository at this point in the history
  • Loading branch information
comfyanonymous committed Nov 11, 2023
1 parent 58d5d71 commit ca2812b
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions comfy_extras/nodes_model_advanced.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit ca2812b

Please sign in to comment.