src/transformers/models/openai/modeling_tf_openai.py/TFOpenAIGPTMainLayer/set_input_embeddings
class TFOpenAIGPTMainLayer:    def set_input_embeddings(self, value):
        self.tokens_embed.weight = value
        self.tokens_embed.vocab_size = shape_list(value)[0]