src/transformers/models/transfo_xl/modeling_transfo_xl.py/TransfoXLModel/set_input_embeddings
class TransfoXLModel:    def set_input_embeddings(self, new_embeddings):
        self.word_emb = new_embeddings