config.json

{
    "vocab_size":100000,
    "n_dim": 1024,
    "n_head": 16,
    "n_kvhead": 4,
    "d_ffn": 4096,
    "m_len": 1024,
    "n_layer": 20,
    "norm_eps": 1e-5
}