zenz-v1-space / conversion.py
Plat
chore: onnx model
f8fa551
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")