config.json

{
    "n_embd": 384,
    "n_heads": 6,
    "n_layers": 6,
    "vocab_size": 2000,
    "block_size": 256
}