How to load pre-trained model to fine-tune it on classification task:
GENA_LM repository
This makes it a practical baseline for genomic modeling experiments and future architectural comparisons.
ModernGENA uses the
32k BPE vocabulary (AIRI-Institute/gena-lm-bert-base-t2t) introduced in GENA-LM, built over the DNA alphabet symbols
A/T/G/C/N, with special tokens
[CLS],
[SEP],
[PAD],
[UNK], and
[MASK].
To reduce overrepresentation of simple repeats and enrich biologically informative regions, training intervals were sampled around transcription start sites:
1from transformers import AutoTokenizer, AutoModel
2
3tokenizer = AutoTokenizer.from_pretrained("AIRI-Institute/gena-lm-bert-base-t2t")
4model = AutoModel.from_pretrained("AIRI-Institute/moderngena-base", trust_remote_code=True, attn_implementation="flash_attention_2")