(self, inputs)
| 250 | return getattr(self.model[0], name) |
| 251 | |
| 252 | def forward(self, inputs): |
| 253 | h, attn_mask, adaln_input, sizes, *freqs_cis = inputs |
| 254 | out = self.final_layer(h, adaln_input) |
| 255 | L_text, gh, gw = sizes |
| 256 | out = -self._tokens_to_img(out[:, L_text:], gh, gw) |
| 257 | return out |
| 258 | |
| 259 | |
| 260 | class Wrapper(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected