Initializes the tokenizer and text encoder model. Args: model_id (str): The model identifier from the Hugging Face Hub.

Args: data_dir (str): Path to the directory containing: - target_crop_info_*.json (metadata file)

Initialize the DMD (Distribution Matching Distillation) module. This class is self-contained and compute generator and fake score los

Initialize the ODERegression module. This class is self-contained and compute generator losses in the forward pass given prec

Initialize the streaming training model. Args: base_model: underlying model (DMD, DMDSwitch, etc.) config: c

Sana training pipeline, refer to SelfForcingTrainingPipeline's interface Args: denoising_step_list: denoising step list

SANA inference pipeline: generate a full video without gradients. The initialization signature is consistent with the use in Trainer

Initialize a new LRU cache with the given capacity.

Create a ShardListDataset. Args: shards: a list of (filename, length) pairs or a URL pointing to a JSON descriptor file

Create a ShardListDataset. Args: shards: a list of (filename, length) pairs or a URL pointing to a JSON descriptor file