hf-transformers-bot's picture
Update tiny models for CpmAntForCausalLM
9d2a600 verified
Raw
History Blame Contribute Delete
542 Bytes
{
"architectures": [
"CpmAntForCausalLM"
],
"dim_ff": 37,
"dim_head": 128,
"dropout_p": 0.0,
"dtype": "float32",
"eps": 1e-06,
"hidden_size": 32,
"init_std": 0.02,
"is_decoder": true,
"model_type": "cpmant",
"num_attention_heads": 4,
"num_hidden_layers": 2,
"position_bias_max_distance": 128,
"position_bias_num_buckets": 32,
"prompt_length": 8,
"prompt_types": 8,
"segment_types": 8,
"tie_word_embeddings": true,
"transformers_version": "5.16.0.dev0",
"use_cache": true,
"vocab_size": 30720
}