Converting Model to ONNX

#10
by kimijx - opened

Hi,

I have been trying to convert this model to onnx using the torch.onnx.export() api, but it is running into a protobuf size limit error.
"The serialized model is larger than the 2GiB limit imposed by the protobuf library. Therefore the output file must be a file path, so that the ONNX external data can be written to the same directory. Please specify the output file name."

I was wondering how the onnx conversion was done for this published model so I can try to follow the same steps also.

Thank you

I was able to convert the onnx model by overwriting the remote code implementation and updating the changes outlined in this PR https://huggingface.co/jinaai/jina-bert-implementation/discussions/7/files

Sign up or log in to comment