src/transformers/models/megatron_bert/modeling_megatron_bert.py/MegatronBertForPreTraining/get_output_embeddings
class MegatronBertForPreTraining:    def get_output_embeddings(self):
        return self.cls.predictions.decoder