OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-02-02 03:32:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #12856 from catboxanon/extra-noise-noisy-latent

Browse Source 
Add noisy latent to `ExtraNoiseParams` for callback
This commit is contained in:
AUTOMATIC1111 2023-08-30 07:27:13 +03:00 committed by GitHub
commit 834f4c7cd3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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