(self)
| 24 | return torch.linspace(beta_start, beta_end, timesteps) |
| 25 | |
| 26 | def quadratic_beta_schedule(self): |
| 27 | timesteps = self.timesteps |
| 28 | scale = 1000 / timesteps |
| 29 | beta_start = 1e-6 * scale |
| 30 | beta_end = 0.02 * scale |
| 31 | return torch.linspace(beta_start ** 0.5, beta_end ** 0.5, timesteps) ** 2 |
| 32 | |
| 33 | def sigmoid_beta_schedule(self): |
| 34 | timesteps = self.timesteps |
nothing calls this directly
no outgoing calls
no test coverage detected