Commit 549b475b authored by catboxanon's avatar catboxanon
Browse files

Add noisy latent to ExtraNoiseParams for callback

parent 04b90328

modules/script_callbacks.py

+5 −2
Original line number Diff line number Diff line
@@ -29,12 +29,15 @@ class ImageSaveParams: 




class ExtraNoiseParams:

    def __init__(self, noise, x):

    def __init__(self, noise, x, xi):

        self.noise = noise

        """Random noise generated by the seed"""



        self.x = x

        """Latent image representation of the image"""

        """Latent representation of the image"""



        self.xi = xi

        """Noisy latent representation of the image"""





class CFGDenoiserParams:

modules/sd_samplers_kdiffusion.py

+1 −1
Original line number Diff line number Diff line
@@ -148,7 +148,7 @@ class KDiffusionSampler(sd_samplers_common.Sampler): 


        if opts.img2img_extra_noise > 0:

            p.extra_generation_params["Extra noise"] = opts.img2img_extra_noise

            extra_noise_params = ExtraNoiseParams(noise, x)

            extra_noise_params = ExtraNoiseParams(noise, x, xi)

            extra_noise_callback(extra_noise_params)

            noise = extra_noise_params.noise

            xi += noise * opts.img2img_extra_noise

modules/sd_samplers_timesteps.py

+1 −1
Original line number Diff line number Diff line
@@ -107,7 +107,7 @@ class CompVisSampler(sd_samplers_common.Sampler): 


        if opts.img2img_extra_noise > 0:

            p.extra_generation_params["Extra noise"] = opts.img2img_extra_noise

            extra_noise_params = ExtraNoiseParams(noise, x)

            extra_noise_params = ExtraNoiseParams(noise, x, xi)

            extra_noise_callback(extra_noise_params)

            noise = extra_noise_params.noise

            xi += noise * opts.img2img_extra_noise * sqrt_alpha_cumprod

CRA Git | Maintained and supported by SUSTech CRA and CCSE