src/transformers/models/megatron_bert/modeling_megatron_bert.py/MegatronBertModel/get_input_embeddings
class MegatronBertModel:    def get_input_embeddings(self):
        return self.embeddings.word_embeddings