Downloads · 30 days
10
2% of all-time downloads
pszemraj/nougat-base-onnx
nougat-base-onnx is a image-text-to-text model from pszemraj. Use it for the image-text-to-text task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as cc-by-4.0.
https://huggingface.co/facebook/nougat-base but exported to onnx. This is not quantized.
Downloads · 30 days
10
2% of all-time downloads
All-time downloads
456
Public
Repo size
2.4 GB
Likes
0
Public
Click a slice to open those files.
.onnx2.4 GB · 100%
From the Hugging Face model README
https://huggingface.co/facebook/nougat-base but exported to onnx. This is not quantized.
from transformers import NougatProcessor
from optimum.onnxruntime import ORTModelForVision2Seq
model_name = 'pszemraj/nougat-base-onnx'
processor = NougatProcessor.from_pretrained(model_name)
model = ORTModelForVision2Seq.from_pretrained(
model_name,
provider="CPUExecutionProvider", # 'CUDAExecutionProvider' for gpu
use_merged=False,
use_io_binding=True
)
on colab CPU-only (at time of writing) you may get CuPy errors, to solve this uninstall it:
pip uninstall cupy-cuda11x -y
See here