(self, labels: torch.LongTensor, force_drop_ids=None)
| 1533 | return labels |
| 1534 | |
| 1535 | def forward(self, labels: torch.LongTensor, force_drop_ids=None): |
| 1536 | use_dropout = self.dropout_prob > 0 |
| 1537 | if (self.training and use_dropout) or (force_drop_ids is not None): |
| 1538 | labels = self.token_drop(labels, force_drop_ids) |
| 1539 | embeddings = self.embedding_table(labels) |
| 1540 | return embeddings |
| 1541 | |
| 1542 | |
| 1543 | class TextImageProjection(nn.Module): |
nothing calls this directly
no test coverage detected