Adaptive Projected Guidance: Shape mismatch fix

2026-06-02 07:08:30 +00:00 · 2026-06-02 07:08:30 +00:00 · 1af8dcacac
parent c96fcddb81
commit 1af8dcacac
1 changed files with 1 additions and 1 deletions
--- a/comfy_extras/nodes_apg.py
+++ b/comfy_extras/nodes_apg.py
@ -73,7 +73,7 @@ class APG(io.ComfyNode):
            guidance = cond - uncond

            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
                else:
                    running_avg = momentum * running_avg + guidance