diff --git a/scripts/img2img.py b/scripts/img2img.py index 421e2151d9..bce5b7d219 100644 --- a/scripts/img2img.py +++ b/scripts/img2img.py @@ -256,7 +256,7 @@ def main(): c = model.get_learned_conditioning(prompts) # encode (scaled latent) - z_enc = sampler.stochastic_encode(init_latent, torch.tensor([t_enc]*batch_size).to(device)) + z_enc = sampler.stochastic_encode(init_latent, torch.tensor([t_enc - 1]*batch_size).to(device)) # decode it samples = sampler.decode(z_enc, c, t_enc, unconditional_guidance_scale=opt.scale, unconditional_conditioning=uc,)