| { | |
| "project_name": "wan8_csquant_patch2_bs8_720p", | |
| "teacher_config": { | |
| "dim": "96", | |
| "z_dim": "16", | |
| "dim_mult": "[1, 2, 4, 4]", | |
| "num_res_blocks": "2", "attn_scales": "[]", "temperal_downsample": "[False, True, True]", "dropout": "0.0", "cls": "<class 'models.temporal_wan.WanVAE_'>" | |
| }, | |
| "student_config": { | |
| "dim": "64", | |
| "z_dim": "16", | |
| "dim_mult": "[1, 2, 4]", | |
| "patch_size": "2", | |
| "num_res_blocks": "3", | |
| "attn_scales": "[]", | |
| "dropout": "0.0", | |
| "cls": "<class 'models.image_vae.DiscreteImageVAE'>", | |
| "z_channels": "256", | |
| "z_factor": "1", | |
| "embedding_dim": "16", | |
| "levels": "[8, 8, 8, 5, 5, 5]", | |
| "dtype": "torch.float32", | |
| "model_type": "wan_2_1", | |
| "quantizer_cls": "<class 'models.quantizers.ChannelSplitFSQ'>", | |
| "num_codebooks": "1", | |
| "K": "2" | |
| } | |
| } |