Module audiocraft.modules

Modules used for building the models.

Sub-modules

audiocraft.modules.activations
audiocraft.modules.chroma
audiocraft.modules.codebooks_patterns
audiocraft.modules.conditioners
audiocraft.modules.conv
audiocraft.modules.diffusion_schedule

Functions for Noise Schedule, defines diffusion process, reverse process and data processor.

audiocraft.modules.jasco_conditioners
audiocraft.modules.lstm
audiocraft.modules.rope
audiocraft.modules.seanet
audiocraft.modules.streaming

Streaming module API that should be implemented by all Streaming components,

audiocraft.modules.transformer

Transformer model, with streaming support, xformer attention support and easy causal attention with a potentially finite receptive field …

audiocraft.modules.unet_transformer
audiocraft.modules.watermark