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