facebookresearch / nougat

Implementation of Nougat Neural Optical Understanding for Academic Documents
https://facebookresearch.github.io/nougat/
MIT License
8.98k stars 567 forks source link

How to switch between "0.1.0-small" and "0.1.0-base? #5

Closed nil-is-lin closed 1 year ago

nil-is-lin commented 1 year ago

I have tried using nougat

nougat .\Nougat- Neural Optical Understanding for Academic Documents.pdf -o ./

It downloaded the 0.1.0-small version. How can I use the 0.1.0-base version for comparison?

lukas-blecher commented 1 year ago

You'll need to manually download the model files from the release section into a separate directory (eg nougat-base) and then provide the path to the model

nougat .\Nougat- Neural Optical Understanding for Academic Documents.pdf -o ./ --checkpoint .\nougat-base