diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-14 05:15:22 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-14 05:15:22 +0000 |
| commit | 005849331e82cded96f6f3e5ff828037c672c38d (patch) | |
| tree | 5f3d9000540e9d63dc9da6887fb4bef97af4877d /modules | |
| parent | cb9a3a7809fc7247360705bca4175ccb59b9288c (diff) | |
| download | stable-diffusion-webui-gfx803-005849331e82cded96f6f3e5ff828037c672c38d.tar.gz stable-diffusion-webui-gfx803-005849331e82cded96f6f3e5ff828037c672c38d.tar.bz2 stable-diffusion-webui-gfx803-005849331e82cded96f6f3e5ff828037c672c38d.zip | |
remove output_altered flag from AfterCFGCallbackParams
Diffstat (limited to 'modules')
| -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
|