| { | |
| "model_type": "complexity-dit", | |
| "architecture": "ComplexityDiT-S", | |
| "d_model": 384, | |
| "num_layers": 12, | |
| "num_heads": 6, | |
| "num_experts": 4, | |
| "context_dim": 768, | |
| "patch_size": 2, | |
| "in_channels": 4, | |
| "image_size": 32, | |
| "dynamics_enabled": true, | |
| "dynamics_alpha": 0.9, | |
| "dynamics_beta": 0.1, | |
| "dynamics_dt": 0.1, | |
| "parameters": "114M", | |
| "training_steps": 20000, | |
| "training_dataset": "huggan/wikiart", | |
| "framework": "pytorch" | |
| } | |