extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-12 21:06:44 +02:00
Code Issues Packages Projects Releases Wiki Activity

img2img attempt for sd2

Browse Source
This commit is contained in:
cmdr2 2022-11-25 22:36:02 +05:30 committed by GitHub
parent 642c114501
commit b924d323d4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 29 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
ui/sd_internal/runtime.py
View File

@ -813,18 +813,35 @@ def _img2img(init_latent, t_enc, batch_size, opt_scale, c, uc, opt_ddim_steps, o
) )
x_T = None if mask is None else init_latent x_T = None if mask is None else init_latent
# decode it if thread_data.test_sd2:
samples_ddim = thread_data.model.sample( from ldm.models.diffusion.ddim import DDIMSampler
t_enc,
c, sampler = DDIMSampler(thread_data.model)
z_enc,
unconditional_guidance_scale=opt_scale, samples_ddim = thread_data.model.sample(
unconditional_conditioning=uc, S=t_enc,
img_callback=img_callback, batch_size=opt_n_samples,
mask=mask, conditioning=c,
x_T=x_T, x0=z_enc,
sampler = 'ddim' unconditional_guidance_scale=opt_scale,
) unconditional_conditioning=uc,
img_callback=img_callback,
mask=mask,
x_T=x_T
)
else:
# decode it
samples_ddim = thread_data.model.sample(
t_enc,
c,
z_enc,
unconditional_guidance_scale=opt_scale,
unconditional_conditioning=uc,
img_callback=img_callback,
mask=mask,
x_T=x_T,
sampler = 'ddim'
)
yield from samples_ddim yield from samples_ddim
def gc(): def gc():