config.json

{"d_model": 1024, "n_layer": 56, "vocab_size": 256, "ssm_cfg": {"expand": 2}, "rms_norm": true, "residual_in_fp32": true, "fused_add_norm": true, "pad_vocab_size_multiple": 8, "tie_embeddings": false, "pad_id": 0}