Struct ESM2 

pub struct ESM2 {
    pub session: SessionBuilder,
    pub model_path: PathBuf,
    pub tokenizer: Tokenizer,
}

Fields

session: SessionBuilder

model_path: PathBuf

tokenizer: Tokenizer

Implementations

impl ESM2

pub fn new(model: ESM2Models) -> Result<Self>

pub fn load_model_path(model: ESM2Models) -> Result<PathBuf>

pub fn load_tokenizer() -> Result<Tokenizer>

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

pub fn extract_logits(&self, tensor: &Tensor) -> Result<Vec<PseudoProbability>>