src/transformers/models/nystromformer/modeling_nystromformer.py/NystromformerModel/set_input_embeddings
class NystromformerModel:    def set_input_embeddings(self, value):
        self.embeddings.word_embeddings = value