extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-08-15 02:39:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix the ddim_timesteps attribute missing error for txt2img with the ddim sampler

Browse Source
This commit is contained in:
cmdr2
2022-09-23 11:14:06 +05:30
parent e7f9db5e56
commit 83cb473a45
2 changed files with 24 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ui/sd_internal/runtime.py
View File

@ -522,6 +522,9 @@ def _txt2img(opt_W, opt_H, opt_n_samples, opt_ddim_steps, opt_scale, start_code,
while torch.cuda.memory_allocated() / 1e6 >= mem:
time.sleep(1)
if sampler_name == 'ddim' and not hasattr(model, 'ddim_timesteps'):
model.make_schedule(ddim_num_steps=opt_ddim_steps, ddim_eta=opt_ddim_eta, verbose=False)
samples_ddim = model.sample(
S=opt_ddim_steps,
conditioning=c,