bert4torch_config / chatglm-6b-v0.1.0 /bert4torch_config.json
Tongjilibo's picture
chatglm中增加generation_config
af50f21
raw
history blame
619 Bytes
{
"model": "glm",
"hidden_act": "gelu_new",
"bos_token_id": 130004,
"eos_token_id": 130005,
"mask_token_id": 130000,
"gmask_token_id": 130001,
"pad_token_id": 3,
"hidden_size": 4096,
"intermediate_size": 16384,
"layer_norm_eps": 1e-05,
"max_sequence_length": 2048,
"num_attention_heads": 32,
"num_hidden_layers": 28,
"position_encoding_2d": true,
"torch_dtype": "float16",
"vocab_size": 130528,
"segment_vocab_size": 0,
"skip_init": true,
"rope_rank": "updown",
"tie_emb_prj_weight": false,
"generation_config": {"max_length": 2048}
}