Fix sampling and scoring device bugs

facebookresearch / esm

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

MIT License

3.16k stars 627 forks source link

Fix sampling and scoring device bugs #488

Closed naailkhan28 closed 1 year ago

naailkhan28 commented 1 year ago

As mentioned in issue #487 there is still some a CUDA device bug in GVPTransformer.model.sample - this PR fixes that plus a bug in esm.inverse_folding.util.get_sequence_loss