This can be removed since PAG doesn't use the uncond.

8903dce8 · comfyanonymous · 719fb2c8 · 8903dce8
Commit 8903dce8 authored Apr 15, 2024 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

comfy_extras/nodes_pag.py comfy_extras/nodes_pag.py +1 -1

No files found.
--- a/comfy_extras/nodes_pag.py
+++ b/comfy_extras/nodes_pag.py
@@ -47,7 +47,7 @@ class PerturbedAttentionGuidance:

            return cfg_result + (cond_pred - pag) * scale

-        m.set_model_sampler_post_cfg_function(post_cfg_function, disable_cfg1_optimization=True)
+        m.set_model_sampler_post_cfg_function(post_cfg_function)

        return (m,)