(self, model)
| 133 | |
| 134 | class InitialLayer(nn.Module): |
| 135 | def __init__(self, model): |
| 136 | super().__init__() |
| 137 | self.input_proj = model.input_proj |
| 138 | self.t_embedding = model.t_embedding |
| 139 | self.adaln_proj = model.adaln_proj |
| 140 | self.llm_cond_norm = model.llm_cond_norm |
| 141 | self.llm_cond_proj = model.llm_cond_proj |
| 142 | self.embed_image_indicator = model.embed_image_indicator |
| 143 | self.model = [model] |
| 144 | |
| 145 | def __getattr__(self, name): |
| 146 | return getattr(self.model[0], name) |