transformers_config.json

{
  "model_type": "tiny_llm",
  "architectures": [
    "TinyLLM"
  ],
  "vocab_size": 74,
  "d_model": 128,
  "num_hidden_layers": 2,
  "num_attention_heads": 2,
  "intermediate_size": 256,
  "max_position_embeddings": 512,
  "pad_token_id": 0
}