fastvideo.v1.dataset.parquet_datasets

Module Contents

Classes

ParquetVideoTextDataset

Efficient loader for video-text data from a directory of Parquet files.

Data

logger

API

class fastvideo.v1.dataset.parquet_datasets.ParquetVideoTextDataset(path: str, batch_size: int = 1024, rank: int = 0, world_size: int = 1, cfg_rate: float = 0.0, num_latent_t: int = 2, seed: int = 0)

Bases: torch.utils.data.Dataset

Efficient loader for video-text data from a directory of Parquet files.

Initialization

fastvideo.v1.dataset.parquet_datasets.logger

‘init_logger(…)’