search

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

Added --model argument for model selection #60

Closed abhimskywalker closed 1 year ago

abhimskywalker commented 1 year ago

This should close: https://github.com/facebookresearch/nougat/issues/59

I tried parsing a few PDFs with small version but it gave MISSING_PAGE on some of the pages, while the base model was able to convert the pages perfectly.

So adding this support for model selection at cli level to help both myself and others. :)

abhimskywalker commented 1 year ago

Closes #59

lukas-blecher commented 1 year ago

Sounds good. I'm a bit worried about backward compatibility though. Maybe instead of redownloading the small model, we can check if the old format is present and then move it to the correct spot?