config.json

{
  "vocab_size": 50000,
  "block_size": 128,
  "n_layer": 6,
  "n_head": 8,
  "n_embd": 256,
  "mlp_ratio": 4,
  "dropout": 0.0,
  "tie_weights": false,
  "model_type": "tinybuddy",
  "architectures": ["TinyGPT"],
  "auto_map": {
    "AutoModelForCausalLM": "modeling_tinybuddy.TinyGPT",
    "AutoConfig": "modeling_tinybuddy.GPTConfig"
  },
  "torch_dtype": "float32"
}