How to export OpenCLIP models to ONNX format?

[pntt3011]

Hi @slavabarkov, I truly appreciate your contribution in making the CLIP model viable on Android devices.

Can I have the code you used to export the OpenCLIP model to ONNX format?

I want to export OpenCLIP's ViT-B-32 datacomp_xl_s13b_b90k model as it has higher accuracy in their benchmarks.

I also tried PicQuery code, so far it worked but the exported model was 10MB bigger than yours.

[junxiant]

Did u find a solution?

[aviv926]

Maybe this can help you

https://github.com/immich-app/immich/tree/main/machine-learning%2Fexport