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

View File

@@ -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