from transformers import AutoConfig, TFAutoModelForCausalLM

model_checkpoint = "gpt2"

config = AutoConfig.from_pretrained(model_checkpoint)

config.save_pretrained("./saved_model/") # 存在这里,然后可以 from_pretrained("./saved_model/")

model = TFAutoModelForCausalLM.from_config(config)