Skip to content

ltx2_precomputed_dataset

Classes

fastvideo.dataset.ltx2_precomputed_dataset.LTX2PrecomputedDataset 

LTX2PrecomputedDataset(data_root: str, data_sources: dict[str, str] | list[str] | None = None)

Bases: Dataset

Dataset for LTX-2 precomputed latents and conditions.

Expected directory structure (data_root): .precomputed/ latents/.pt conditions/.pt audio_latents/*.pt (optional)

Source code in fastvideo/dataset/ltx2_precomputed_dataset.py 
40
41
42
43
44
45
46
47
48
49
50
def __init__(
    self,
    data_root: str,
    data_sources: dict[str, str] | list[str] | None = None,
) -> None:
    super().__init__()
    self.data_root = self._setup_data_root(data_root)
    self.data_sources = self._normalize_data_sources(data_sources)
    self.source_paths = self._setup_source_paths()
    self.sample_files = self._discover_samples()
    self._validate_setup()

Functions