BALM-paired is an antibody language model that uses a
RoBERTa architecture and was pre-trained on the ~1.6M paired antibody sequences from
Jaffe et al.
Datasets used for pre-training are available on
Zenodo and code is available on
GitHub. More details can be found in
our paper published in Patterns.
1from transformers import RobertaTokenizer, RobertaForMaskedLM
2
3model = RobertaForMaskedLM.from_pretrained("brineylab/BALM-paired")
4tokenizer = RobertaTokenizer.from_pretrained("brineylab/BALM-paired")