From 005849331e82cded96f6f3e5ff828037c672c38d Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sun, 14 May 2023 08:15:22 +0300
Subject: [PATCH] remove output_altered flag from AfterCFGCallbackParams

---
 modules/script_callbacks.py       | 3 ---
 modules/sd_samplers_kdiffusion.py | 3 +--
 2 files changed, 1 insertion(+), 5 deletions(-)

diff --git a/modules/script_callbacks.py b/modules/script_callbacks.py
index 57dfd457..3c21a362 100644
--- a/modules/script_callbacks.py
+++ b/modules/script_callbacks.py
@@ -78,9 +78,6 @@ class AfterCFGCallbackParams:
         self.total_sampling_steps = total_sampling_steps
         """Total number of sampling steps planned"""
 
-        self.output_altered = False
-        """A flag for CFGDenoiser indicating whether the output has been altered by the callback"""
-
 
 class UiTrainTabParams:
     def __init__(self, txt2img_preview_params):
diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py
index 55f0d3a3..61f23ad7 100644
--- a/modules/sd_samplers_kdiffusion.py
+++ b/modules/sd_samplers_kdiffusion.py
@@ -183,8 +183,7 @@ class CFGDenoiser(torch.nn.Module):
 
         after_cfg_callback_params = AfterCFGCallbackParams(denoised, state.sampling_step, state.sampling_steps)
         cfg_after_cfg_callback(after_cfg_callback_params)
-        if after_cfg_callback_params.output_altered:
-            denoised = after_cfg_callback_params.x
+        denoised = after_cfg_callback_params.x
 
         self.step += 1
         return denoised