(self, x_t, t, noise)
| 233 | return mean, variance, log_variance |
| 234 | |
| 235 | def predict_start_from_noise(self, x_t, t, noise): |
| 236 | return ( |
| 237 | extract(self.sqrt_recip_alphas_cumprod, t, x_t.shape) * x_t - |
| 238 | extract(self.sqrt_recipm1_alphas_cumprod, t, x_t.shape) * noise |
| 239 | ) |
| 240 | |
| 241 | def q_posterior(self, x_start, x_t, t): |
| 242 | posterior_mean = ( |
no test coverage detected