src/transformers/models/gpt2/modeling_tf_gpt2.py/TFGPT2MainLayer/set_input_embeddings
class TFGPT2MainLayer:    def set_input_embeddings(self, value):
        self.wte.weight = value
        self.wte.vocab_size = shape_list(value)[0]