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