(self, config)
| 1391 | ) |
| 1392 | class LlamaForSequenceClassification(LlamaPreTrainedModel): |
| 1393 | def __init__(self, config): |
| 1394 | super().__init__(config) |
| 1395 | self.num_labels = config.num_labels |
| 1396 | self.model = LlamaModel(config) |
| 1397 | self.score = nn.Linear(config.hidden_size, self.num_labels, bias=False) |
| 1398 | |
| 1399 | # Initialize weights and apply final processing |
| 1400 | self.post_init() |
| 1401 | |
| 1402 | def get_input_embeddings(self): |
| 1403 | return self.model.embed_tokens |
nothing calls this directly
no test coverage detected