diff options
author | ritosonn <ritosonn@gmail.com> | 2022-10-21 14:46:32 +0000 |
---|---|---|
committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-25 05:32:33 +0000 |
commit | b38370275275bf6e11575000f39c50c6e90b1f7a (patch) | |
tree | ad2f2e5bffae984afcf4964692aed09b79da02ed /modules/sd_samplers.py | |
parent | 73c4a8138cd547a40337053d78a6efb4d9627352 (diff) | |
download | stable-diffusion-webui-gfx803-b38370275275bf6e11575000f39c50c6e90b1f7a.tar.gz stable-diffusion-webui-gfx803-b38370275275bf6e11575000f39c50c6e90b1f7a.tar.bz2 stable-diffusion-webui-gfx803-b38370275275bf6e11575000f39c50c6e90b1f7a.zip |
fix #3145 #3093
Diffstat (limited to 'modules/sd_samplers.py')
-rw-r--r-- | modules/sd_samplers.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index 0b408a70..3670b57d 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -228,7 +228,7 @@ class VanillaStableDiffusionSampler: unconditional_conditioning = {"c_concat": [image_conditioning], "c_crossattn": [unconditional_conditioning]}
- samples = self.launch_sampling(steps, lambda: self.sampler.decode(x1, conditioning, t_enc, unconditional_guidance_scale=p.cfg_scale, unconditional_conditioning=unconditional_conditioning))
+ samples = self.launch_sampling(t_enc + 1, lambda: self.sampler.decode(x1, conditioning, t_enc, unconditional_guidance_scale=p.cfg_scale, unconditional_conditioning=unconditional_conditioning))
return samples
@@ -429,7 +429,7 @@ class KDiffusionSampler: self.model_wrap_cfg.init_latent = x
self.last_latent = x
- samples = self.launch_sampling(steps, lambda: self.func(self.model_wrap_cfg, xi, extra_args={
+ samples = self.launch_sampling(t_enc + 1, lambda: self.func(self.model_wrap_cfg, xi, extra_args={
'cond': conditioning,
'image_cond': image_conditioning,
'uncond': unconditional_conditioning,
|