Can I export any hugging face checkpoint to onnx? If not, how do I go about it @nielsr I want to export trocr checkpoint to onnx, is it possible.
I tried doing the same with a fine-tuned checkpoint of mine, it gives a KEY ERROR
I’ve read through this documentation about implementing a custom ONNX configuration for unsupported architectures. The walk-through described here doesn’t look too difficult. But, I decided that before I attempted this I’d ask: is this possible with TrOCR in its current state, or is this irrelevant until the code is refactored (which I know you smart people are working on currently, and I thank you for it)?
We’re currently refactoring the internals of the ONNX exporter to handle multimodal models better here. Once that’s done, it should be relatively straightforward to export TrOCR and I’d be happy to guide you on that