(self)
| 17 | return torch.clip(betas, 0.0001, 0.9999) |
| 18 | |
| 19 | def linear_beta_schedule(self): |
| 20 | timesteps = self.timesteps |
| 21 | scale = 1000 / timesteps |
| 22 | beta_start = 1e-6 * scale |
| 23 | beta_end = 0.02 * scale |
| 24 | return torch.linspace(beta_start, beta_end, timesteps) |
| 25 | |
| 26 | def quadratic_beta_schedule(self): |
| 27 | timesteps = self.timesteps |