File size: 225 Bytes
f8fa551
 
 
 
 
 
 
1
2
3
4
5
6
7
8
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")