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.

Save the features and statistics to 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

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

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

Find and clean up any ComfyUI temp directories

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

Returns the names of available models

Dynamically find the ComfyUI temp directory

Initializes distributed environment.

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

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

Initializes and returns the dataloader.

Initializes and return the dataset.

Validate int64 integers and convert negative integers to positive by backward search