holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Ddim decode for img2img

Browse Source
This commit is contained in:
cmdr2 2022-11-26 00:55:39 +05:30 committed by GitHub
parent 6a8985d8dd
commit e503c6092e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
ui/sd_internal/runtime.py
View File

@ -819,19 +819,8 @@ def _img2img(init_latent, t_enc, batch_size, opt_scale, c, uc, opt_ddim_steps, o
z_enc = sampler.stochastic_encode(init_latent, torch.tensor([t_enc] * batch_size).to(thread_data.device)) z_enc = sampler.stochastic_encode(init_latent, torch.tensor([t_enc] * batch_size).to(thread_data.device))
#samples = sampler.decode(z_enc, c, t_enc, unconditional_guidance_scale=opt.scale,unconditional_conditioning=uc, ) samples_ddim = sampler.decode(z_enc, c, t_enc, unconditional_guidance_scale=opt_scale,unconditional_conditioning=uc, img_callback=img_callback)
samples_ddim = sampler.sample(
t_enc,
batch_size,
shape = [opt_C, opt_H // opt_f, opt_W // opt_f],
conditioning=c,
x0=z_enc,
unconditional_guidance_scale=opt_scale,
unconditional_conditioning=uc,
img_callback=img_callback,
mask=mask,
x_T=x_T
)
else: else:
z_enc = thread_data.model.stochastic_encode( z_enc = thread_data.model.stochastic_encode(
init_latent, init_latent,