Commit 00584933 authored by AUTOMATIC's avatar AUTOMATIC

remove output_altered flag from AfterCFGCallbackParams

parent cb9a3a78
......@@ -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):
......
......@@ -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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment