facebook

/

nougat-small

like 27

Follow

AI at Meta 5.51k

Here's one exported to onnx https://huggingface.co/pszemraj/nougat-small-onnx.
There are three onnx model files after export: decoder_model.onnx, decoder_with_past_model.onnx, encoder_model.onnx.
However the author doesn't provide the details how he did that.

torch.onnx.export only export one model file model.onnx, and ORTModelForVision2Seq connot load it:

FileNotFoundError: Could not find any ONNX model file for the regex ['(.*)?decoder((?!(with_past|merged)).)*?\\.onnx'] in /home/zhichyu/model_dir/nougat.