(model, steps, shape, x0)
| 93 | |
| 94 | |
| 95 | def convsample_ddim_forward(model, steps, shape, x0): |
| 96 | with model.ema_scope("Plotting"): # TODO: important |
| 97 | ddim = DDIMSampler(model) |
| 98 | bs = shape[0] |
| 99 | shape = shape[1:] |
| 100 | samples, intermediates = ddim.sample_forward(steps, batch_size=bs, shape=shape, eta=0, verbose=True, x0=x0) # TODO: verbose. |
| 101 | return samples, intermediates |
| 102 | |
| 103 | |
| 104 | def convsample_ddim_forward_conditional(model, steps, shape, x0, scale, class_label): |
no test coverage detected