(self, model)
| 122 | |
| 123 | class InitialLayer(nn.Module): |
| 124 | def __init__(self, model): |
| 125 | super().__init__() |
| 126 | self.t_embedder = model.t_embedder |
| 127 | self.cap_embedder = model.cap_embedder |
| 128 | self.rope_embedder = model.rope_embedder |
| 129 | self.context_refiner = model.context_refiner |
| 130 | self.x_embedder = model.x_embedder |
| 131 | self.noise_refiner = model.noise_refiner |
| 132 | self.model = [model] |
| 133 | |
| 134 | def __getattr__(self, name): |
| 135 | return getattr(self.model[0], name) |