model_config.json

{
  "vocab_size": 88,
  "dim": 32,
  "num_heads": 2,
  "num_layers": 2,
  "max_len": 32
}