xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-05 06:07:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix modified cond to match ref impl better, support different cond arity

Browse Source
This commit is contained in:
thot-experiment 2025-05-12 18:05:24 -07:00
parent 20cf3d36e0
commit 0c98ae69ca
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy_extras/nodes_apg.py
View File

@ -28,9 +28,12 @@ class APG:
def pre_cfg_function(args): def pre_cfg_function(args):
nonlocal running_avg, prev_sigma nonlocal running_avg, prev_sigma
if len(args["conds_out"]) == 1: return args["conds_out"]
cond = args["conds_out"][0] cond = args["conds_out"][0]
uncond = args["conds_out"][1] uncond = args["conds_out"][1]
sigma = args["sigma"][0] sigma = args["sigma"][0]
cond_scale = args["cond_scale"]
if prev_sigma is not None and sigma > prev_sigma: if prev_sigma is not None and sigma > prev_sigma:
running_avg = 0 running_avg = 0
@ -56,9 +59,9 @@ class APG:
guidance_parallel, guidance_orthogonal = project(guidance, cond) guidance_parallel, guidance_orthogonal = project(guidance, cond)
modified_guidance = guidance_orthogonal + eta * guidance_parallel modified_guidance = guidance_orthogonal + eta * guidance_parallel
modified_cond = uncond + modified_guidance modified_cond = (uncond + modified_guidance) + (cond - uncond) / cond_scale
return [modified_cond, uncond] return [modified_cond, uncond] + args["conds_out"][2:]
m = model.clone() m = model.clone()
m.set_model_sampler_pre_cfg_function(pre_cfg_function) m.set_model_sampler_pre_cfg_function(pre_cfg_function)