(self, x_t, t, noise)
| 289 | return mean, variance, log_variance |
| 290 | |
| 291 | def predict_start_from_noise(self, x_t, t, noise): |
| 292 | return ( |
| 293 | extract_into_tensor(self.sqrt_recip_alphas_cumprod, t, x_t.shape) * x_t - |
| 294 | extract_into_tensor(self.sqrt_recipm1_alphas_cumprod, t, x_t.shape) * noise |
| 295 | ) |
| 296 | |
| 297 | def predict_start_from_z_and_v(self, x_t, t, v): |
| 298 | # self.register_buffer('sqrt_alphas_cumprod', to_torch(np.sqrt(alphas_cumprod))) |
no test coverage detected