from transformers import AutoModelForCausalLM

model.save_pretrained("./my_model")