(self, new_num_tokens)
| 401 | return model_embeds |
| 402 | |
| 403 | def _resize_token_embeddings(self, new_num_tokens): |
| 404 | old_embeddings = self.get_input_embeddings() |
| 405 | new_embeddings = self._get_resized_embeddings(old_embeddings, new_num_tokens) |
| 406 | self.set_input_embeddings(new_embeddings) |
| 407 | return self.get_input_embeddings() |
| 408 | |
| 409 | def _get_resized_embeddings( |
| 410 | self, old_embeddings: torch.nn.Embedding, new_num_tokens: Optional[int] = None |
no test coverage detected