wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-15 08:40:50 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Merge 0783a02d9f into 19f2192d69

Browse Source
This commit is contained in:
Extraltodeus 2025-12-03 11:43:28 -05:00 committed by GitHub
commit 1eb04ec004
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/samplers.py
View File

@ -355,8 +355,10 @@ def cfg_function(model, cond_pred, uncond_pred, cond_scale, x, timestep, model_o
args = {"cond": x - cond_pred, "uncond": x - uncond_pred, "cond_scale": cond_scale, "timestep": timestep, "input": x, "sigma": timestep,
"cond_denoised": cond_pred, "uncond_denoised": uncond_pred, "model": model, "model_options": model_options, "input_cond": cond, "input_uncond": uncond}
cfg_result = x - model_options["sampler_cfg_function"](args)
else:
elif uncond_pred.any():
cfg_result = uncond_pred + (cond_pred - uncond_pred) * cond_scale
else:
cfg_result = cond_pred
for fn in model_options.get("sampler_post_cfg_function", []):
args = {"denoised": cfg_result, "cond": cond, "uncond": uncond, "cond_scale": cond_scale, "model": model, "uncond_denoised": uncond_pred, "cond_denoised": cond_pred,