(self, *args, **kwargs)
| 1683 | |
| 1684 | @torch.no_grad() |
| 1685 | def log_images(self, *args, **kwargs): |
| 1686 | log = super(LatentInpaintDiffusion, self).log_images(*args, **kwargs) |
| 1687 | log["masked_image"] = rearrange(args[0]["masked_image"], |
| 1688 | 'b h w c -> b c h w').to(memory_format=torch.contiguous_format).float() |
| 1689 | return log |
| 1690 | |
| 1691 | |
| 1692 | class LatentDepth2ImageDiffusion(LatentFinetuneDiffusion): |
nothing calls this directly
no test coverage detected