Enum ESM2Models 

pub enum ESM2Models {
    T6_8M,
    T12_35M,
    T30_150M,
    T33_650M,
    T36_3B,
    T48_15B,
    Esm1v1,
    Esm1v2,
    Esm1v3,
    Esm1v4,
    Esm1v5,
    Esm1b,
    SaProt35M,
    SaProt650M,
    FastEsm2_650,
    PepMlm650M,
    Dplm650M,
}

Variants

T6_8M

T12_35M

T30_150M

T33_650M

T36_3B

T48_15B

Esm1v1

ESM-1v UR90S ensemble member 1 (ferritin-goh.4).

Esm1v2

ESM-1v UR90S ensemble member 2 (ferritin-goh.4).

Esm1v3

ESM-1v UR90S ensemble member 3 (ferritin-goh.4).

Esm1v4

ESM-1v UR90S ensemble member 4 (ferritin-goh.4).

Esm1v5

ESM-1v UR90S ensemble member 5 (ferritin-goh.4).

Esm1b

ESM-1b — learned absolute positions.

SaProt35M

SaProt 35M — ESM-2 architecture, structure-aware alphabet (ferritin-goh.3).

SaProt650M

SaProt 650M — the variant most people use.

FastEsm2_650

FastESM2 650M — ESM-2 650M republished as safetensors (ferritin-goh.12).

PepMlm650M

PepMLM 650M — ESM-2 650M fine-tuned for peptide binder design.

Dplm650M

DPLM 650M — a discrete diffusion model on ESM-2’s architecture.

Its hidden states are ordinary embeddings; its lm_head denoises rather than scoring masked tokens, so do not read its logits as masked-LM scores.

Implementations

impl ESM2Models

pub const fn registry_id(&self) -> &'static str

This variant’s registry id.

pub fn card(&self) -> &'static ModelCard

This variant’s ModelCard.

pub fn model_info(&self) -> (WeightSource, ESM2Config)

Where this variant’s weights live, plus its built-in fallback config.

The weight source comes from REGISTRY rather than being repeated here (ferritin-goh.1). The config stays a separate hardcoded fallback because it is only used when the hub’s config.json is unreachable or unparseable.

pub fn get_model_files(model: Self) -> (WeightSource, ESM2Config)

👎Deprecated since 0.4.0:

renamed to model_info, which takes &self

Renamed to model_info (ferritin-100.8).