diff options
| -rw-r--r-- | modules/script_callbacks.py | 3 | ||||
| -rw-r--r-- | modules/sd_samplers_kdiffusion.py | 3 |
2 files changed, 1 insertions, 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
|