(self, config: TelechatConfig)
| 743 | _keys_to_ignore_on_load_missing = [r"lm_head.weight"] |
| 744 | |
| 745 | def __init__(self, config: TelechatConfig): |
| 746 | super().__init__(config) |
| 747 | self.transformer = TelechatModel(config) |
| 748 | self.lm_head = nn.Linear(config.hidden_size, config.vocab_size, bias=False) |
| 749 | self.post_init() |
| 750 | |
| 751 | def get_output_embeddings(self): |
| 752 | return self.lm_head |
nothing calls this directly
no test coverage detected