src/transformers/models/realm/modeling_realm.py/RealmKnowledgeAugEncoder/set_input_embeddings
class RealmKnowledgeAugEncoder:    def set_input_embeddings(self, value):
        self.realm.embeddings.word_embeddings = value