biotransformers.bio_transformers

Contents

`biotransformers.bio_transformers`¶

Main module to build either ESM or protbert model

Module Contents¶

Classes¶

BioTransformers

General class to choose an ESM or ProtBert backend

class biotransformers.bio_transformers.BioTransformers(backend: str = 'esm1_t6_43M_UR50S', num_gpus: int = 0)¶

Bases: biotransformers.wrappers.transformers_wrappers.TransformersWrapper

General class to choose an ESM or ProtBert backend Abstract method are implemented in transformers

static list_backend() → None¶: Get all possible backend for the model