View Source API Reference Bumblebee v0.4.2

Modules

Pre-trained Axon models for easy inference and boosted training.

High-level tasks related to audio processing.

Whisper model family.

Whisper featurizer for audio data.

An interface for configurable entities.

Denoising diffusion implicit models (DDIMs).

Pseudo numerical methods for diffusion models (PNDMs).

High-level tasks based on Stable Diffusion.

A CLIP-based model for detecting unsafe image content.

U-Net model with two spatial dimensions and conditional state.

Variational autoencoder (VAE) with Kullback–Leibler divergence (KL) loss.

An interface for configuring and applying featurizers.

An interface for configuring and building models based on the same architecture.

The BLIP model for text-image similarity.

The CLIP model for text-image similarity.

LayoutLM Model family.

An interface for configuring and using schedulers.

High-level tasks related to text processing.

ALBERT model family.

ALBERT tokenizer.

BART model family.

BERT model family.

Blenderbot model family.

Blenderbot tokenizer.

The BLIP model for text encoding.

Camembert tokenizer.

The CLIP model for text encoding.

DistilBERT model family.

DistilBERT tokenizer.

An interface for language models supporting sequence generation.

A set of configuration options controlling text generation.

GPT-2 model family.

GPT-2 tokenizer.

GPT-NeoX model family.

GPT-NeoX tokenizer.

LayoutLM tokenizer.

LLaMA model family.

mBART model family.

RoBERTa model family.

RoBERTa tokenizer.

T5 model family.

A set of Whisper-specific configuration options controlling text generation.

Whisper tokenizer.

XLM-RoBERTa tokenizer.

An interface for configuring and applying tokenizers.

High-level tasks related to vision.

BLIP featurizer for image data.

The BLIP model for image encoding.

CLIP featurizer for image data.

The CLIP model for image encoding.

ConvNeXT model family.

ConvNeXT featurizer for image data.

DeiT model family.

DeiT featurizer for image data.

ResNet model family.

ViT model family.

ViT featurizer for image data.