(self, transformer, spatial_compression_factor)
| 295 | |
| 296 | class InitialLayer(nn.Module): |
| 297 | def __init__(self, transformer, spatial_compression_factor): |
| 298 | super().__init__() |
| 299 | self.transformer = [transformer] |
| 300 | self.spatial_compression_factor = spatial_compression_factor |
| 301 | self.x_embedder = transformer.x_embedder |
| 302 | self.extra_pos_embedder = transformer.extra_pos_embedder |
| 303 | self.pos_embedder = transformer.pos_embedder |
| 304 | self.t_embedder = transformer.t_embedder |
| 305 | |
| 306 | def __getattr__(self, name): |
| 307 | return getattr(self.transformer[0], name) |