(self, model)
| 169 | |
| 170 | class InitialLayer(nn.Module): |
| 171 | def __init__(self, model): |
| 172 | super().__init__() |
| 173 | self.pos_embed = model.pos_embed |
| 174 | self.time_text_embed = model.time_text_embed |
| 175 | self.context_embedder = model.context_embedder |
| 176 | self.model = [model] |
| 177 | |
| 178 | def __getattr__(self, name): |
| 179 | return getattr(self.model[0], name) |