diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-09-28 19:30:52 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-28 19:30:52 +0000 |
| commit | d62954c2bc149053f9f51dfe95751b9e0ea29f03 (patch) | |
| tree | 03683dd83d0b4e22897f213e2e1aa8cbf5553713 /modules/sd_samplers.py | |
| parent | a703f56d73fa7f9bc3dc6181c3235187027edb01 (diff) | |
| download | stable-diffusion-webui-gfx803-d62954c2bc149053f9f51dfe95751b9e0ea29f03.tar.gz stable-diffusion-webui-gfx803-d62954c2bc149053f9f51dfe95751b9e0ea29f03.tar.bz2 stable-diffusion-webui-gfx803-d62954c2bc149053f9f51dfe95751b9e0ea29f03.zip | |
fix broken DDIM with img2img
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 3588aae6..fc0c94b4 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -136,6 +136,8 @@ class VanillaStableDiffusionSampler: def sample_img2img(self, p, x, noise, conditioning, unconditional_conditioning, steps=None):
steps, t_enc = setup_img2img_steps(p, steps)
+ self.initialize(p)
+
# existing code fails with cetain step counts, like 9
try:
self.sampler.make_schedule(ddim_num_steps=steps, ddim_eta=self.eta, ddim_discretize=p.ddim_discretize, verbose=False)
@@ -144,8 +146,6 @@ class VanillaStableDiffusionSampler: x1 = self.sampler.stochastic_encode(x, torch.tensor([t_enc] * int(x.shape[0])).to(shared.device), noise=noise)
- self.initialize(p)
-
self.init_latent = x
self.step = 0
|