From 8903dce862616ddd937400ffe6a971b66409262f Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Mon, 15 Apr 2024 12:14:00 -0400
Subject: [PATCH] This can be removed since PAG doesn't use the uncond.

---
 comfy_extras/nodes_pag.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy_extras/nodes_pag.py b/comfy_extras/nodes_pag.py
index c48a3958..63f43fd6 100644
--- 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,)