(self, x: torch.Tensor, t: torch.Tensor, **kwargs)
| 102 | self.ddim_sampler = DDIMSampler(self.diffusion) |
| 103 | |
| 104 | def forward(self, x: torch.Tensor, t: torch.Tensor, **kwargs): |
| 105 | return self.net(x, t, **kwargs) |
| 106 | |
| 107 | def training_losses(self, x1: torch.Tensor, x0: torch.Tensor = None, **cond_kwargs): |
| 108 | loss, _ = self.diffusion.training_losses( |
nothing calls this directly
no outgoing calls
no test coverage detected