src/transformers/models/mbart/modeling_mbart.py/MBartForConditionalGeneration/get_output_embeddings
class MBartForConditionalGeneration:    def get_output_embeddings(self):
        return self.lm_head