(self, processed_images, tiled=False, tile_size=64, tile_stride=32)
| 131 | |
| 132 | |
| 133 | def encode_video(self, processed_images, tiled=False, tile_size=64, tile_stride=32): |
| 134 | latents = [] |
| 135 | for image in processed_images: |
| 136 | image = self.preprocess_image(image).to(device=self.device, dtype=self.torch_dtype) |
| 137 | latent = self.encode_image(image, tiled=tiled, tile_size=tile_size, tile_stride=tile_stride) |
| 138 | latents.append(latent.cpu()) |
| 139 | latents = torch.concat(latents, dim=0) |
| 140 | return latents |
| 141 | |
| 142 | |
| 143 | @torch.no_grad() |
no test coverage detected