config.json

{
    "d_model": 1024,
    "n_layer": 48,
    "vocab_size": 50304,
    "ssm_cfg": {},
    "rms_norm": true,
    "residual_in_fp32": true,
    "fused_add_norm": true,
    "pad_vocab_size_multiple": 128
}