add Apple Silicon support

mheinzinger / ProstT5

Bilingual Language Model for Protein Sequence and Structure

MIT License

176 stars 15 forks source link

Closed gbouras13 closed 6 months ago

gbouras13 commented 6 months ago

Adds support for Apple Silicon GPUs in these scripts - I have tested it out on a Mac M1 Pro chip and it works well, should work generally.

mheinzinger commented 6 months ago

Nice, thanks! :)