diff options
| author | 不会画画的中医不是好程序员 <yfszzx@gmail.com> | 2022-10-25 15:38:33 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-25 15:38:33 +0800 |
| commit | 5bfa2b23ca6dad7d2664bfab5dab4b1dfabf5b7f (patch) | |
| tree | 7e23ff3f180ada71bc0824270efa82e7cfe68397 /modules/sd_samplers.py | |
| parent | ff305acd51cc71c5eea8aee0f537a26a6d1ba2a1 (diff) | |
| parent | 91c1e1e6a92061b99c92a5b1d548535907d2ad96 (diff) | |
Merge branch 'AUTOMATIC1111:master' into Inspiron
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,
|