Skip to main content

PlmRunner

ferritin_plms::plm_runner

Trait PlmRunner 

Source 
pub trait PlmRunner {
    // Required methods
    fn embed(&self, sequence: &str) -> Result<Tensor>;
    fn model_name(&self) -> &str;
}
Expand description

Trait implemented by all PLM runner types.

Required Methods§

Source

fn embed(&self, sequence: &str) -> Result<Tensor>

Run a forward pass on sequence and return per-residue embeddings.

Shape: (1, L, d_model) where L includes any BOS/EOS tokens.

Source

fn model_name(&self) -> &str

Model name / identifier string (e.g. “esm2”, “amplify”, “esmc”).

Implementors§