(self, s_churn=0.0, s_tmin=0.0, s_tmax=float("inf"), s_noise=1.0, *args, **kwargs)
| 84 | |
| 85 | class EDMSampler(SingleStepDiffusionSampler): |
| 86 | def __init__(self, s_churn=0.0, s_tmin=0.0, s_tmax=float("inf"), s_noise=1.0, *args, **kwargs): |
| 87 | super().__init__(*args, **kwargs) |
| 88 | |
| 89 | self.s_churn = s_churn |
| 90 | self.s_tmin = s_tmin |
| 91 | self.s_tmax = s_tmax |
| 92 | self.s_noise = s_noise |
| 93 | |
| 94 | def sampler_step(self, sigma, next_sigma, denoiser, x, cond, uc=None, gamma=0.0): |
| 95 | sigma_hat = sigma * (gamma + 1.0) |