(self, inputs)
| 485 | |
| 486 | @torch.autocast('cuda', dtype=AUTOCAST_DTYPE) |
| 487 | def forward(self, inputs): |
| 488 | hidden_states, llama_encoder_hidden_states, adaln_input, rope, image_tokens_seq_len = inputs |
| 489 | hidden_states = hidden_states[:, :image_tokens_seq_len.item(), ...] |
| 490 | return self.final_layer(hidden_states, adaln_input) |
nothing calls this directly
no outgoing calls
no test coverage detected