298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-03-15 14:09:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix RescaleCFG for batch size > 1.

Browse Source
This commit is contained in:
comfyanonymous 2023-11-10 22:05:25 -05:00
parent 58d5d71a93
commit ca2812bae0
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy_extras/nodes_model_advanced.py
View File

@ -140,6 +140,7 @@ class RescaleCFG:
uncond = args["uncond"] uncond = args["uncond"]
cond_scale = args["cond_scale"] cond_scale = args["cond_scale"]
sigma = args["sigma"] sigma = args["sigma"]
sigma = sigma.view(sigma.shape[:1] + (1,) * (cond.ndim - 1))
x_orig = args["input"] x_orig = args["input"]
#rescale cfg has to be done on v-pred model output #rescale cfg has to be done on v-pred model output