Mirrors
/
stable-diffusion-webui
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add noisy latent to ExtraNoiseParams for callback

This commit is contained in:
catboxanon 2023-08-29 14:22:04 -04:00
parent 04b90328c0
commit 549b475be9
3 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
modules/script_callbacks.py
View File

@ -29,12 +29,15 @@ class ImageSaveParams:
class ExtraNoiseParams: class ExtraNoiseParams:
def __init__(self, noise, x): def __init__(self, noise, x, xi):
self.noise = noise self.noise = noise
"""Random noise generated by the seed""" """Random noise generated by the seed"""
self.x = x 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: class CFGDenoiserParams:

2
modules/sd_samplers_kdiffusion.py
View File

@ -148,7 +148,7 @@ class KDiffusionSampler(sd_samplers_common.Sampler):
if opts.img2img_extra_noise > 0: if opts.img2img_extra_noise > 0:
p.extra_generation_params["Extra noise"] = opts.img2img_extra_noise 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) extra_noise_callback(extra_noise_params)
noise = extra_noise_params.noise noise = extra_noise_params.noise
xi += noise * opts.img2img_extra_noise xi += noise * opts.img2img_extra_noise

2
modules/sd_samplers_timesteps.py
View File

@ -107,7 +107,7 @@ class CompVisSampler(sd_samplers_common.Sampler):
if opts.img2img_extra_noise > 0: if opts.img2img_extra_noise > 0:
p.extra_generation_params["Extra noise"] = opts.img2img_extra_noise 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) extra_noise_callback(extra_noise_params)
noise = extra_noise_params.noise noise = extra_noise_params.noise
xi += noise * opts.img2img_extra_noise * sqrt_alpha_cumprod xi += noise * opts.img2img_extra_noise * sqrt_alpha_cumprod