Repository: micanonsens/bigamba-seq_only-step44000
The Gamba models are a family of DNA language models from the ArGamba paper that jointly model DNA sequence and evolutionary rate information.
This repository includes the model weights and code for the Caduceus-based bidirectional variant model (BiGamba) trained to predict the human genome sequence using the Masked Language Modelling (MLM) task. For more details, see the GitHub repo.
All Gamba family models have checkpoints available at 44,000 steps:
| Checkpoint name | Architecture | Training task |
|---|---|---|
| ArGamba-dual | ArGamba (Jamba autoregressive) | NTP + CEP |
| ArGamba-seq_only | ArGamba (Jamba autoregressive) | NTP |
| ArGamba-cons_only | ArGamba (Jamba autoregressive) | CEP |
| BiGamba-dual | BiGamba (Mamba bidirectional) | MLM + MEM |
| BiGamba-seq_only | BiGamba (Mamba bidirectional) | MLM |
| BiGamba-cons_only | BiGamba (Mamba bidirectional) | MEM |
from transformers import AutoModel
model = AutoModel.from_pretrained(
"micanonsens/bigamba-seq_only-step44000",
trust_remote_code=True
)
trust_remote_code=True is required.Repository: micanonsens/bigamba-seq_only-step44000
The Gamba models are a family of DNA language models from the ArGamba paper that jointly model DNA sequence and evolutionary rate information.
This repository includes the model weights and code for the Caduceus-based bidirectional variant model (BiGamba) trained to predict the human genome sequence using the Masked Language Modelling (MLM) task. For more details, see the GitHub repo.
All Gamba family models have checkpoints available at 44,000 steps:
| Checkpoint name | Architecture | Training task |
|---|---|---|
| ArGamba-dual | ArGamba (Jamba autoregressive) | NTP + CEP |
| ArGamba-seq_only | ArGamba (Jamba autoregressive) | NTP |
| ArGamba-cons_only | ArGamba (Jamba autoregressive) | CEP |
| BiGamba-dual | BiGamba (Mamba bidirectional) | MLM + MEM |
| BiGamba-seq_only | BiGamba (Mamba bidirectional) | MLM |
| BiGamba-cons_only | BiGamba (Mamba bidirectional) | MEM |
from transformers import AutoModel
model = AutoModel.from_pretrained(
"micanonsens/bigamba-seq_only-step44000",
trust_remote_code=True
)
trust_remote_code=True is required.