search

agemagician / ProtTrans

ProtTrans is providing state of the art pretrained language models for proteins. ProtTrans was trained on thousands of GPUs from Summit and hundreds of Google TPUs using Transformers Models.
Academic Free License v3.0
1.13k stars 153 forks source link

Fix full precision casting on cpu #139

Closed li-yq closed 10 months ago

li-yq commented 10 months ago

I get the same error as #130 in the prott5_embedder.py script. Code was edited according to the colab notebook.

PS: I'm still learning about PyTorch, so please correct me if anything goes wrong.

mheinzinger commented 10 months ago

Good catch; thanks!