config.json

{
  "vocab_size": 128, 
  "n_embd": 128,
  "block_size": 64,
  "n_layer": 8,
  "n_head": 8,
  "model_type": "custom_transformer"
}