LayerNorm but with an optional bias. PyTorch doesn't support simply bias=False

Block class represents a single block in the model. Args: config (object): Configuration object containing parameters for the block.

Base class for audio enhancers

Mixin class for performing inference in a causal language model. This mixin provides methods for predicting the next token in a sequence, sa

Mixin class for non-causal inference in a language model. This class provides methods for performing non-causal sampling using a language mo

Sample from a trained model.