SAVRN
Search Contact SAVRN

Open-weight model · Fill mask

esm2_t6_8M_UR50D

by AI at Meta facebook/esm2_t6_8M_UR50D

ESM-2 is a state-of-the-art protein model trained on a masked language modelling objective. It is suitable for fine-tuning on a wide range of tasks that take protein sequences as input.

Parameters8M
Context1,026
Weights93.0 MB
Licensemit
AccessOpen weights
Monthly Downloads608.8k

Runs On

What it takes to serve esm2_t6_8M_UR50D (8M parameters): the memory its weights need at each precision, and the cheapest way to rent enough data-center GPUs to hold them.

PrecisionWeightsMemory neededCheapest setupPer hourAlso fits
16-bit 0.0 GB 0.0 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
8-bit 0.0 GB 0.0 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
4-bit 0.0 GB 0.0 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00

Memory is the weights at that precision plus 20% for the runtime and a short context; a long context needs more. Prices are the lowest on-demand hourly rates in the SAVRN Index, read Sep 18, 2026.

SAVRN's Notes on esm2_t6_8M_UR50D

A vocabulary of 33 tokens tells you what this model reads: amino acid sequences, not prose. Meta trained it with a masked language modeling objective and positions it as a base for fine-tuning on tasks that take protein sequences as input. At 8 million parameters across six layers, the weights round to 0.0 GB at 16-bit, the nine files total about 93 MB, and our cheapest listing is one MI300X at $1.85 per hour on-demand. It rides in the margin of hardware a lab already runs.

MIT terms allow commercial use, modification and redistribution with the notices intact, which covers an in-house fine-tune and a derived model shipped to customers. Check the 1,026 token context against your sequence lengths. The publisher notes that larger checkpoints generally have somewhat better accuracy but need much more memory, so this one trades accuracy for a footprint close to free.

Model Card

By AI at Meta, published under mit, revision c731040fcd8d.

ESM-2 is a state-of-the-art protein model trained on a masked language modelling objective. It is suitable for fine-tuning on a wide range of tasks that take protein sequences as input. For detailed information on the model architecture and training data, please refer to the accompanying paper. You may also be interested in some demo notebooks (PyTorch, TensorFlow) which demonstrate how to fine-tune ESM-2 models on your tasks of interest. Several ESM-2 checkpoints are available in the Hub with varying sizes. Larger sizes generally have somewhat better accuracy, but require much more memory and time to train

Read AI at Meta's full model card

ESM-2

ESM-2 is a state-of-the-art protein model trained on a masked language modelling objective. It is suitable for fine-tuning on a wide range of tasks that take protein sequences as input. For detailed information on the model architecture and training data, please refer to the accompanying paper. You may also be interested in some demo notebooks (PyTorch, TensorFlow) which demonstrate how to fine-tune ESM-2 models on your tasks of interest.

Several ESM-2 checkpoints are available in the Hub with varying sizes. Larger sizes generally have somewhat better accuracy, but require much more memory and time to train:

Checkpoint name Num layers Num parameters
esm2_t48_15B_UR50D 48 15B
esm2_t36_3B_UR50D 36 3B
esm2_t33_650M_UR50D 33 650M
esm2_t30_150M_UR50D 30 150M
esm2_t12_35M_UR50D 12 35M
esm2_t6_8M_UR50D 6 8M

Configuration

Architecture
EsmForMaskedLM
Context length (tokens)
1,026
Layers
6
Hidden size
320
Feed-forward size
1,280
Attention heads
20
Vocabulary size
33
Stored precision
float32
Model type
esm

Identity and Version

Repository
facebook/esm2_t6_8M_UR50D
Publisher
AI at Meta
Task
Fill mask
Modality
Text
Library
transformers
Parameters
8M parameters
Languages
tf, esm
Revision
c731040fcd8d73dceaa04b0a8e6329b345b0f5df
First published
2022-09-26
Last updated
2023-03-21

Files and Weights

9 files, 93.1 MB in total. The weights are 3 files totalling 93.0 MB in bin, h5, safetensors.

Weights3 files · 93.0 MB
Configuration2 files · 900 B
Tokenizer2 files · 188 B
Documentation1 file · 1.7 KB
Repository1 file · 1.4 KB
Every file
FileTypeSizeSHA-256
model.safetensorsWeights31.4 MB 24c5fa474c48
pytorch_model.binWeights31.4 MB 9edcf393212f
tf_model.h5Weights30.3 MB bbb600f46992
config.jsonConfiguration775 B
special_tokens_map.jsonConfiguration125 B
README.mdDocumentation1.7 KB
.gitattributesRepository1.4 KB
tokenizer_config.jsonTokenizer95 B
vocab.txtTokenizer93 B

License and Download

License
mit
Access
Open weights, no gate
Download size
93.0 MB
Download from AI at Meta

Released by AI at Meta through its official repository on Hugging Face. Read the license.

Memory Requirements

PrecisionWeights in memory
As published93.0 MB
16-bit0.0 GB
8-bit0.0 GB
4-bit0.0 GB

Weights only, from the published parameter count; the key-value cache and runtime add to this.

Questions About esm2_t6_8M_UR50D

How much GPU memory does esm2_t6_8M_UR50D need?

About 0 GB at 16-bit and 0 GB at 4-bit: the weights (8M parameters) plus a working margin. A long context needs more.

What is the cheapest GPU to run esm2_t6_8M_UR50D on?

At 16-bit, 1x MI300X from $1.85 an hour; at 4-bit, 1x MI300X from $1.85 an hour, at the lowest on-demand prices the SAVRN Index lists.

Can I use esm2_t6_8M_UR50D commercially?

Yes. esm2_t6_8M_UR50D is released under MIT License. The MIT License is a short permissive license. It permits commercial use, modification and redistribution, provided the copyright notice and permission notice are included.

What is esm2_t6_8M_UR50D's context length?

1,026 tokens, from the maximum position embeddings in its published configuration.

Similar Models

To use the pre-trained model for masked language modeling, use the following snippet: Alternatively, you can instantiate a model from scratch to train on your own data as follows: This is the Caduceus-PS model with hidden dimension 256 and 16 MambaDNA layers. This model is reverse complement (RC) equivariant and thus no RC data augmentation is required when training this model, either during pre-training or for downstream fine-tuning. Note that the model hidden state will be twice that of a non-RC equivariant counterpart. For downstream task training and inference, and to ensure RC invariant outputs at downstream time, one can either run the downstream model on the hidden state and its RC…

Open weights apache-2.0 8M parameters transformers

To use the pre-trained model for masked language modeling, use the following snippet: Alternatively, you can instantiate a model from scratch to train on your own data as follows: This is the Caduceus-Ph model with hidden dimension 256 and 16 MambaDNA layers. This model is not inherently reverse complement (RC) equivariant. Rather, it was pre-trained using RC data augmentation. Its intended usage is as follows: for downstream tasks, the model should be trained with RC data augmentation. At downstream task inference, the model should be run twice: once on a sequence and once on its RC. The output of these two applications should be combined (averaged) to form the downstream task prediction.…

Open weights apache-2.0 8M parameters transformers

Model · Fill mask

albert-base-v2

ALBERT community

Pretrained model on English language using a masked language modeling (MLM) objective. It was introduced in this paper and first released in this repository. This model, as all ALBERT models, is uncased: it does not make a difference between english and English. Disclaimer: The team releasing ALBERT did not write a model card for this model so this model card has been written by the Hugging Face team. ALBERT is a transformers model pretrained on a large corpus of English data in a self-supervised fashion. This means it was pretrained on the raw texts only, with no humans labelling them in any way (which is why it can use lots of publicly available data) with an automatic process to generate…

Open weights apache-2.0 12M parameters 512 tokens transformers

Model · Fill mask

mdeberta-v3-base

Microsoft

DeBERTa improves the BERT and RoBERTa models using disentangled attention and enhanced mask decoder. With those two improvements, DeBERTa out perform RoBERTa on a majority of NLU tasks with 80GB training data. In DeBERTa V3, we further improved the efficiency of DeBERTa using ELECTRA-Style pre-training with Gradient Disentangled Embedding Sharing. Compared to DeBERTa, our V3 version significantly improves the model performance on downstream tasks. You can find more technique details about the new model from our paper. Please check the official repository for more implementation details and updates. mDeBERTa is multilingual version of DeBERTa which use the same structure as DeBERTa and was…

Open weights mit 512 tokens transformers

Model · Fill mask

deberta-v3-base

Microsoft

DeBERTa improves the BERT and RoBERTa models using disentangled attention and enhanced mask decoder. With those two improvements, DeBERTa out perform RoBERTa on a majority of NLU tasks with 80GB training data. In DeBERTa V3, we further improved the efficiency of DeBERTa using ELECTRA-Style pre-training with Gradient Disentangled Embedding Sharing. Compared to DeBERTa, our V3 version significantly improves the model performance on downstream tasks. You can find more technique details about the new model from our paper. Please check the official repository for more implementation details and updates. The DeBERTa V3 base model comes with 12 layers and a hidden size of 768. It has only 86M…

Open weights mit 512 tokens transformers

Model · Fill mask

Bio_ClinicalBERT

Emily Alsentzer

The Publicly Available Clinical BERT Embeddings paper contains four unique clinicalBERT models: initialized with BERT-Base (casedL-12H-768A-12) or BioBERT (BioBERT-Base v1.0 + PubMed 200K + PMC 270K) & trained on either all MIMIC notes or only discharge summaries. This model card describes the Bio+Clinical BERT model, which was initialized from BioBERT & trained on all MIMIC notes. The BioClinicalBERT model was trained on all notes from MIMIC III, a database containing electronic health records from ICU patients at the Beth Israel Hospital in Boston, MA. For more details on MIMIC, see here. All notes from the NOTEEVENTS table were included (~880M words). Each note in MIMIC was first split…

Open weights mit 512 tokens transformers