(self, latents, return_dict=None)
| 286 | self.config = DotDict(scaling_factor=1.0, latents_mean=torch.zeros(16), z_dim=16, latents_std=torch.ones(16)) |
| 287 | |
| 288 | def decode(self, latents, return_dict=None): |
| 289 | n, c, t, h, w = latents.shape |
| 290 | return (self.taehv.decode_video(latents.transpose(1, 2), parallel=False).transpose(1, 2).mul_(2).sub_(1),) |
| 291 | |
| 292 | def stream_decode_with_cond(self, latents, tiled=False, cond=None): |
| 293 | n, c, t, h, w = latents.shape |
nothing calls this directly
no test coverage detected