from optimum.onnxruntime import ORTModelForCausalLM

MODEL_NAME = "Miwa-Keita/zenz-v1-checkpoints"

ort_model = ORTModelForCausalLM.from_pretrained(MODEL_NAME, export=True)

ort_model.save_pretrained(save_directory="./onnx")