| { |
| "model": { |
| "model_paths": "Models/Wan2.2-TI2V-5B" |
| }, |
| "dataset": { |
| "dataset_repeat": 1, |
| "dataset_num_workers": 8, |
| "data_file_keys": "video" |
| }, |
| "action": { |
| "action_type": "state_pose" |
| }, |
| "video": { |
| "height": 480, |
| "width": 640, |
| "spatial_division_factor": 32, |
| "max_pixels": 16777216, |
| "num_frames": 81, |
| "num_history_frames": 9 |
| }, |
| "training": { |
| "learning_rate": 5e-05, |
| "seed": 42, |
| "max_train_steps": 12000, |
| "trainable_models": "dit", |
| "weight_decay": 0.01, |
| "task": "sft", |
| "mixed_precision": "bf16", |
| "max_timestep_boundary": 1.0, |
| "min_timestep_boundary": 0.0 |
| }, |
| "infer": { |
| "cfg_scale": 1.0, |
| "num_inference_steps": 50, |
| "quality": 5, |
| "fps": 30 |
| } |
| } |
