| from transformers import AutoModelForCausalLM, AutoTokenizer | |
| model_name = "gpt2" | |
| model = AutoModelForCausalLM.from_pretrained(model_name) | |
| tokenizer = AutoTokenizer.from_pretrained(model_name) | |
| model.save_pretrained("./my_model") | |
| tokenizer.save_pretrained("./my_model") | |