config.json

{
  "hidden_size": 128,
  "num_heads": 4,
  "mlp_ratio": 4.0,
  "num_blocks": 4,
  "grid_size": 9
}