Returns the model's input embeddings. Returns: :obj:`nn.Module`: A torch module mapping vocabulary to hidden states.
(self)
| 308 | return getattr(self, self.base_model_prefix, self) |
| 309 | |
| 310 | def get_input_embeddings(self): |
| 311 | """ |
| 312 | Returns the model's input embeddings. |
| 313 | |
| 314 | Returns: |
| 315 | :obj:`nn.Module`: |
| 316 | A torch module mapping vocabulary to hidden states. |
| 317 | """ |
| 318 | base_model = getattr(self, self.base_model_prefix, self) |
| 319 | if base_model is not self: |
| 320 | return base_model.get_input_embeddings() |
| 321 | else: |
| 322 | raise NotImplementedError |
| 323 | |
| 324 | def set_input_embeddings(self, value: nn.Module): |
| 325 | """ |
no outgoing calls
no test coverage detected