(self, x_start, t, noise=None)
| 361 | return_intermediates=return_intermediates) |
| 362 | |
| 363 | def q_sample(self, x_start, t, noise=None): |
| 364 | noise = default(noise, lambda: torch.randn_like(x_start)) |
| 365 | return (extract_into_tensor(self.sqrt_alphas_cumprod, t, x_start.shape) * x_start + |
| 366 | extract_into_tensor(self.sqrt_one_minus_alphas_cumprod, t, x_start.shape) * noise) |
| 367 | |
| 368 | def get_v(self, x, noise, t): |
| 369 | return ( |
no test coverage detected