| { | |
| "engine_config": { | |
| "load": "C:/models/Agnes-Video-2.0-MoE/dit", | |
| "cp_size": 1 | |
| }, | |
| "arch_config": { | |
| "sparse_moe_layers": [ | |
| 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, | |
| 29, | |
| 30, | |
| 31, | |
| 32, | |
| 33, | |
| 34, | |
| 35 | |
| ], | |
| "num_routed_experts": 3, | |
| "moe_top_k": 1, | |
| "moe_aux_loss_coef": 0.01, | |
| "moe_router_noise": 0.0, | |
| "token_sparse_moe_convert_mm_mlp": false, | |
| "mm_layers": [ | |
| 0, | |
| 1, | |
| 2, | |
| 3, | |
| 36, | |
| 37, | |
| 38, | |
| 39 | |
| ], | |
| "audio_in_channels": 128 | |
| }, | |
| "evaluation_config": { | |
| "fps": 30, | |
| "cfg_number": 2, | |
| "num_inference_steps": 32, | |
| "audio_vae_backend": "omnivae", | |
| "omnivae_audio_ckpt_dir": "C:/models/OmniVAE/audio_decoder", | |
| "audio_model_path": "", | |
| "txt_model_path": "C:/models/T5Gemma", | |
| "vae_model_path": "C:/models/Wan2.2-TI2V-5B", | |
| "use_turbo_vae": false, | |
| "student_config_path": "", | |
| "student_ckpt_path": "" | |
| } | |
| } | |