torchcell.models.llm.PeptideModel

class PeptideModel(model_name: str)

Bases: ABC

Abstract interface for peptide-sequence language models.

property max_sequence_size: int

Returns the maximum sequence size for the transformer model.

abstractmethod load_model(model_name: str) → None

Load the tokenizer and model weights for the given model name.

abstractmethod embed(sequences: list[str], mean_embedding: bool = False) → Tensor

Return embeddings for the given sequences.