bert4torch_config / chatglm3-6b /bert4torch_config.json
Tongjilibo's picture
chatglm中增加generation_config
af50f21
{
"model": "glm2",
"hidden_act": "swiglu",
"hidden_size": 4096,
"intermediate_size": 13696,
"layer_norm_eps": 1e-05,
"max_sequence_length": 32768,
"num_attention_heads": 32,
"num_hidden_layers": 28,
"vocab_size": 65024,
"segment_vocab_size": 0,
"multi_query_group_num": 2,
"skip_init": true,
"tie_emb_prj_weight": false,
"eos_token_id": 2,
"pad_token_id": 0,
"rmsnorm": true,
"rope_rank": "adjacent",
"position_encoding_2d": true,
"flash_attention": true,
"generation_config": {"end_id": [2, 64795, 64797], "max_length": 8192}
}