(self, x_t, t, noise)
| 267 | """ Sampling """ |
| 268 | |
| 269 | def predict_start_from_noise(self, x_t, t, noise): |
| 270 | return ( |
| 271 | extract_into_tensor(self.sqrt_recip_alphas_cumprod, t, x_t.shape) * x_t - |
| 272 | extract_into_tensor(self.sqrt_recipm1_alphas_cumprod, t, x_t.shape) * noise |
| 273 | ) |
| 274 | |
| 275 | def predict_start_from_z_and_v(self, x_t, t, v): |
| 276 | return ( |
no test coverage detected