rs239 / ablm

Protein language model customized for antibodies
MIT License
62 stars 8 forks source link

ESM-2 Clarification #3

Closed arolihas closed 1 year ago

arolihas commented 1 year ago

Which ESM-2 model was used for models/AbMAP_esm2_H.pt and L.pt? I could not tell from the paper.

arolihas commented 1 year ago

Based on ablm/embed.py it seems to be esm2_t33_650M_UR50D since it has the same embedding dimension of 1280, just wanted to confirm though.

lordim commented 1 year ago

That's correct. It has the same embedding dimension of 1280 as ESM-1b. So it is esm2_t33_650M_UR50D.