src/transformers/models/xlnet/modeling_tf_xlnet.py/TFXLNetMainLayer/set_input_embeddings
class TFXLNetMainLayer:    def set_input_embeddings(self, value):
        self.word_embedding.weight = value
        self.word_embedding.vocab_size = shape_list(value)[0]