(self, inputs)
| 539 | |
| 540 | @torch.autocast('cuda', dtype=AUTOCAST_DTYPE) |
| 541 | def forward(self, inputs): |
| 542 | hidden_states, encoder_hidden_states, temb, freqs_cos, freqs_sin, img_seq_len = inputs |
| 543 | img_seq_len = img_seq_len[0].item() |
| 544 | hidden_states = hidden_states[:, :img_seq_len, ...] |
| 545 | hidden_states = self.norm_out(hidden_states, temb) |
| 546 | return self.proj_out(hidden_states) |
nothing calls this directly
no outgoing calls
no test coverage detected