aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_samplers_cfg_denoiser.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-08-12 09:39:59 +0000
committerAUTOMATIC1111 <16777216c@gmail.com>2023-08-12 09:39:59 +0000
commit64311faa6848d641cc452115e4e1eb47d2a7b519 (patch)
tree3849ad4ca8ad7f44c8f20e3ab359578f3a4021ec /modules/sd_samplers_cfg_denoiser.py
parent26c92f056acc795af5066779f1b8aedb8dfa983d (diff)
downloadstable-diffusion-webui-gfx803-64311faa6848d641cc452115e4e1eb47d2a7b519.tar.gz
stable-diffusion-webui-gfx803-64311faa6848d641cc452115e4e1eb47d2a7b519.tar.bz2
stable-diffusion-webui-gfx803-64311faa6848d641cc452115e4e1eb47d2a7b519.zip
put refiner into main UI, into the new accordions section
add VAE from main model into infotext, not from refiner model option to make scripts UI without gr.Group fix inconsistencies with refiner when usings samplers that do more denoising than steps
Diffstat (limited to 'modules/sd_samplers_cfg_denoiser.py')
-rw-r--r--modules/sd_samplers_cfg_denoiser.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/modules/sd_samplers_cfg_denoiser.py b/modules/sd_samplers_cfg_denoiser.py
index a532e013..113425b2 100644
--- a/modules/sd_samplers_cfg_denoiser.py
+++ b/modules/sd_samplers_cfg_denoiser.py
@@ -45,6 +45,11 @@ class CFGDenoiser(torch.nn.Module):
self.nmask = None
self.init_latent = None
self.steps = None
+ """number of steps as specified by user in UI"""
+
+ self.total_steps = None
+ """expected number of calls to denoiser calculated from self.steps and specifics of the selected sampler"""
+
self.step = 0
self.image_cfg_scale = None
self.padded_cond_uncond = False
@@ -56,7 +61,6 @@ class CFGDenoiser(torch.nn.Module):
def inner_model(self):
raise NotImplementedError()
-
def combine_denoised(self, x_out, conds_list, uncond, cond_scale):
denoised_uncond = x_out[-uncond.shape[0]:]
denoised = torch.clone(denoised_uncond)