M1/M2 Macs - Githubissues

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

Closed austinmw closed 1 year ago

austinmw commented 1 year ago

Hi, could you please update to add a check for Mac Metal Performance Shader?

device = "mps" if torch.backends.mps.is_available() else "cpu"

lukas-blecher commented 1 year ago

Duplicate of #61