config.json

{
  "model_type": "tinyllm",
  "vocab_size": 8000,
  "d_model": 512,
  "n_head": 8,
  "n_layer": 8,
  "d_ff": 2048,
  "block_size": 512,
  "tie_word_embeddings": true
}