Add the newly computed features to the list. Update the mean and covariance. Args: x: New features to record.

Load the features and statistics from a pickle file. Args: pkl_file: Path to the pickle file.

Specify how to select the next token, based on the current trace and logits Parameters ---------- tokens : Tensor, shape = (n

Save the features and statistics to a pickle file. Args: pkl_file: Path to the pickle file.

Apply any filtering or masking to logits in-place Parameters ---------- logits : Tensor, shape = (n_batch, vocab_size)

Get entire batch of audio samples and video frames. Parameters ---------- indices : list of integers A list of fr

Sinusoid position encoding table

Get sliced features based on a given index :param feature_array: :param start_idx: the start index of the feature :pa

Returns the names of available models

Initializes distributed environment.

Load a Whisper ASR model Parameters ---------- name : str one of the official model names listed by `whisper.available_model

Perform a forward pass on the decoder and return per-token logits

r"""Fills the input Tensor with values drawn from a truncated normal distribution. The values are effectively drawn from the normal distributi

Initializes and returns the dataloader.

Initializes and return the dataset.