No support for Apple Silicon Devices

[Ruchir3003]

There is currently no support for Apple Silicon Devices. If the team could add a check for mps devices, that would be great for Apple users

[ebetica]

We're not planning to support apple silicon, if it's something you need and have some idea of how to do this, please send in a PR!

[imranq2]

Here's the PR: https://github.com/evolutionaryscale/esm/pull/99

PyTorch already supports Mac (MPS): https://pytorch.org/docs/stable/notes/mps.html

Note that MPS does not support the embedding operations so the following environment variable has to be set to allow PyTorch to fallback for those operations: export PYTORCH_ENABLE_MPS_FALLBACK=1

@Ruchir3003 if you want to try it, you can replace the esm in your requirements.txt with this: git+https://github.com/imranq2/esm.git@fix-for-mac-silicon#egg=esm