| """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", | |
| ] | |