Views
No views yet
1from transformers import RobertaTokenizer, RobertaForMaskedLM
2
3model = RobertaForMaskedLM.from_pretrained("brineylab/BALM-unpaired")
4tokenizer = RobertaTokenizer.from_pretrained("brineylab/BALM-unpaired")HEAVY_CHAIN or LIGHT_CHAIN.