(self, s_noise=0.1, *args, **kwargs)
| 127 | |
| 128 | class DDIMSampler(SingleStepDiffusionSampler): |
| 129 | def __init__(self, s_noise=0.1, *args, **kwargs): |
| 130 | super().__init__(*args, **kwargs) |
| 131 | |
| 132 | self.s_noise = s_noise |
| 133 | |
| 134 | def sampler_step(self, sigma, next_sigma, denoiser, x, cond, uc=None, s_noise=0.0): |
| 135 | denoised = self.denoise(x, denoiser, sigma, cond, uc) |