torchcell.models.llm.NucleotideModel

class NucleotideModel(model_name: str)

Bases: ABC

Abstract interface for nucleotide-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.