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