wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-16 11:28:54 +08:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

Merge 1af8dcacac into 091b70edda

Browse Source
This commit is contained in:
TheToxin-git 2026-07-08 17:39:46 +00:00 committed by GitHub
commit 74eaa96ae1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy_extras/nodes_apg.py
View File

@ -73,7 +73,7 @@ class APG(io.ComfyNode):
guidance = cond - uncond guidance = cond - uncond
if momentum != 0: if momentum != 0:
if not torch.is_tensor(running_avg): if not torch.is_tensor(running_avg) or (running_avg.shape != guidance.shape):
running_avg = guidance running_avg = guidance
else: else:
running_avg = momentum * running_avg + guidance running_avg = momentum * running_avg + guidance