Commit 00584933 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

remove output_altered flag from AfterCFGCallbackParams

parent cb9a3a78

modules/script_callbacks.py

+0 −3
Original line number Original line Diff line number Diff line
@@ -78,9 +78,6 @@ class AfterCFGCallbackParams: 
        self.total_sampling_steps = total_sampling_steps

        self.total_sampling_steps = total_sampling_steps

        """Total number of sampling steps planned"""

        """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:

class UiTrainTabParams:

    def __init__(self, txt2img_preview_params):

    def __init__(self, txt2img_preview_params):

modules/sd_samplers_kdiffusion.py

+1 −2
Original line number Original line Diff line number Diff line
@@ -183,7 +183,6 @@ class CFGDenoiser(torch.nn.Module): 




        after_cfg_callback_params = AfterCFGCallbackParams(denoised, state.sampling_step, state.sampling_steps)

        after_cfg_callback_params = AfterCFGCallbackParams(denoised, state.sampling_step, state.sampling_steps)

        cfg_after_cfg_callback(after_cfg_callback_params)

        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

        self.step += 1

CRA Git | Maintained and supported by SUSTech CRA and CCSE