| 290 | |
| 291 | @dataclass |
| 292 | class InferenceConfig: |
| 293 | video_tokenizer_path: Optional[str] |
| 294 | latent_actions_path: Optional[str] |
| 295 | dynamics_path: Optional[str] |
| 296 | device: str |
| 297 | generation_steps: int |
| 298 | context_window: int |
| 299 | fps: int |
| 300 | temperature: float |
| 301 | use_actions: bool |
| 302 | teacher_forced: bool |
| 303 | use_latest_checkpoints: bool |
| 304 | prediction_horizon: int |
| 305 | dataset: str |
| 306 | use_gt_actions: bool |
| 307 | # Inference performance options |
| 308 | amp: bool |
| 309 | tf32: bool |
| 310 | compile: bool |
| 311 | # Interactive mode (user enters action ids) |
| 312 | use_interactive_mode: bool |
| 313 | preload_ratio: Optional[float] = None |
| 314 | |
| 315 | |
| 316 | def load_config(config_cls, default_config_path: Optional[str] = None): |
nothing calls this directly
no outgoing calls
no test coverage detected