{
    "d_model": 512,
    "num_heads": 8,
    "num_kv_heads": 8,
    "num_layers": 8,
    "ff_dim": 2048,
    "batch_size": 70,
    "freeze": [0, 1],
    "num_workers": 1
}