config.json

{
  "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"
}