(model, steps, shape, eta=1.0
)
| 65 | |
| 66 | |
| 67 | def convsample_ddim(model, steps, shape, eta=1.0 |
| 68 | ): |
| 69 | ddim = DDIMSampler(model) |
| 70 | bs = shape[0] |
| 71 | shape = shape[1:] |
| 72 | samples, intermediates = ddim.sample(steps, batch_size=bs, shape=shape, eta=eta, verbose=False,) |
| 73 | return samples, intermediates |
| 74 | |
| 75 | |
| 76 | def make_convolutional_sample(model, batch_size, vanilla=False, custom_steps=None, eta=1.0,): |
no test coverage detected