File size: 635 Bytes
510ab6b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | """Offline data utilities for the Self-Forcing Predictor v4."""
from .capture import PredictorV4TeacherCapture
from .schema import (
CANDIDATE_BLOCK_IDS,
CHUNK_FRAMES,
NUM_CHUNKS,
NUM_STEPS,
SCHEMA_VERSION,
validate_case_tensors,
validate_clean_prefeature,
validate_step_tensors,
)
from .writer import PredictorV4DatasetWriter
__all__ = [
"CANDIDATE_BLOCK_IDS",
"CHUNK_FRAMES",
"NUM_CHUNKS",
"NUM_STEPS",
"SCHEMA_VERSION",
"PredictorV4DatasetWriter",
"PredictorV4TeacherCapture",
"validate_case_tensors",
"validate_clean_prefeature",
"validate_step_tensors",
]
|