{
  "_class_name": "ProMoETransformer2DModel",
  "architecture": "promoe_tc",
  "model_config": {
    "MoE_config": {
      "init_MoeMLP": false,
      "interleave": true,
      "moe_intermediate_size": 1536,
      "num_routed_experts": 12,
      "shared_expert_intermediate_size": 1536,
      "top_k": 1,
      "use_shared_expert": true,
      "use_uncond_expert": true
    },
    "depth": 12,
    "hidden_size": 768,
    "input_size": 32,
    "num_classes": 1000,
    "num_heads": 12,
    "patch_size": 2
  }
}