Custom model export to onnx-runtime

I think this might be linked to this issue.
Could you try installing optimum from sources and re-run your script?