src/transformers/models/megatron_bert/modeling_megatron_bert.py/MegatronBertModel/set_input_embeddings
class MegatronBertModel:    def set_input_embeddings(self, value):
        self.embeddings.word_embeddings = value