src/transformers/models/squeezebert/modeling_squeezebert.py/SqueezeBertModel/set_input_embeddings
class SqueezeBertModel:    def set_input_embeddings(self, new_embeddings):
        self.embeddings.word_embeddings = new_embeddings