DCLM-1B-v0 / config.json
sedrickkeh's picture
Update config.json
35ec1ec verified
raw
history blame
871 Bytes
{
"architectures": [
"OpenLMModel"
],
"model_type": "openlm",
"params": null,
"apply_qk_norm": true,
"attn_activation": null,
"attn_name": "auto",
"attn_seq_scalar": null,
"attn_seq_scalar_alpha": null,
"dim": 2048,
"ffn_type": "swiglu_torch",
"model": "open_lm_1b_swiglutorch",
"model_norm": "gain_only_lp_layer_norm",
"moe_capacity_factor": 1.25,
"moe_expert_model_parallelism": false,
"moe_freq": 0,
"moe_loss_weight": 0.1,
"moe_num_experts": null,
"moe_top_k": 2,
"moe_weight_parallelism": false,
"n_heads": 16,
"n_layers": 24,
"norm_eps": 1e-05,
"positional_embedding_type": "rotary",
"post_embed_norm": false,
"qk_norm": true,
"seq_len": 2048,
"vocab_size": 50432,
"weight_tying": false,
"torch_dtype": "float32",
"transformers_version": "4.40.2"
}