(self, x_t, t, v)
| 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))) |
| 299 | # self.register_buffer('sqrt_one_minus_alphas_cumprod', to_torch(np.sqrt(1. - alphas_cumprod))) |
| 300 | return ( |
| 301 | extract_into_tensor(self.sqrt_alphas_cumprod, t, x_t.shape) * x_t - |
| 302 | extract_into_tensor(self.sqrt_one_minus_alphas_cumprod, t, x_t.shape) * v |
| 303 | ) |
| 304 | |
| 305 | def predict_eps_from_z_and_v(self, x_t, t, v): |
| 306 | return ( |
no test coverage detected