Update runtime.py

This commit is contained in:
cmdr2 2022-11-25 23:36:43 +05:30 committed by GitHub
parent 29484867ca
commit a1d75d40aa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -822,7 +822,7 @@ def _img2img(init_latent, t_enc, batch_size, opt_scale, c, uc, opt_ddim_steps, o
#samples = sampler.decode(z_enc, c, t_enc, unconditional_guidance_scale=opt.scale,unconditional_conditioning=uc, ) #samples = sampler.decode(z_enc, c, t_enc, unconditional_guidance_scale=opt.scale,unconditional_conditioning=uc, )
samples_ddim = sampler.sample( samples_ddim = sampler.sample(
S=t_enc, S=t_enc,
batch_size=opt_n_samples, batch_size=batch_size,
conditioning=c, conditioning=c,
x0=z_enc, x0=z_enc,
unconditional_guidance_scale=opt_scale, unconditional_guidance_scale=opt_scale,