(self, x_t, t, eps)
| 336 | } |
| 337 | |
| 338 | def _predict_xstart_from_eps(self, x_t, t, eps): |
| 339 | assert x_t.shape == eps.shape |
| 340 | return ( |
| 341 | _extract_into_tensor(self.sqrt_recip_alphas_cumprod, t, x_t.shape) * x_t |
| 342 | - _extract_into_tensor(self.sqrt_recipm1_alphas_cumprod, t, x_t.shape) * eps |
| 343 | ) |
| 344 | |
| 345 | def _predict_eps_from_xstart(self, x_t, t, pred_xstart): |
| 346 | return ( |
no test coverage detected