Allow inverse folding model to score sequences on GPU

facebookresearch / esm

Evolutionary Scale Modeling (esm): Pretrained language models for proteins

MIT License

3.26k stars 643 forks source link

Allow inverse folding model to score sequences on GPU #317

Closed hnisonoff closed 1 year ago

hnisonoff commented 2 years ago

I modified the get_sequence_loss function to use whichever device the model is currently on to also be the device that the batch of sequences are put on.