| { | |
| "model_type": "router", | |
| "architecture": "dit", | |
| "num_clusters": 8, | |
| "hidden_size": 768, | |
| "num_layers": 12, | |
| "num_heads": 12, | |
| "patch_size": 2, | |
| "in_channels": 4, | |
| "use_dit_time_embed": true | |
| } | |
| { | |
| "model_type": "router", | |
| "architecture": "dit", | |
| "num_clusters": 8, | |
| "hidden_size": 768, | |
| "num_layers": 12, | |
| "num_heads": 12, | |
| "patch_size": 2, | |
| "in_channels": 4, | |
| "use_dit_time_embed": true | |
| } | |