Skip to content

GitLab

"tools/git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "1f6d3149edc3f2d1c5e5a3829d37153f14d1dbd4"
Commit ca2812ba authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix RescaleCFG for batch size > 1.

parent 58d5d71a
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
comfy_extras/nodes_model_advanced.py
View file @ ca2812ba
...@@ -140,6 +140,7 @@ class RescaleCFG: ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment