{
    "n_embd": 384,
    "n_head": 6,
    "n_out": 1,
    "depth": 96,
    "enable_adj_emb": false,
    "enable_diffusion": true,
    "n_edge_feat": 1,
    "edge_feat_vocab_size": 220,
    "task_level": "e",
    "n_decode_layers": 1
}