config.json

{
    "dimension": 6,
    "n_heads": 2,
    "n_layers": 8
}