CodVa-1-Small / config.json
Bc-AI's picture
Update config.json
873ad8e verified
Raw
History Blame Contribute Delete
871 Bytes
{
"model_type": "codva1",
"architectures": ["Codva1ForCausalLM"],
"auto_map": {
"AutoConfig": "configuration_codva1.CodVa1Config",
"AutoModelForCausalLM": "modeling_codva1.CodVa1ForCausalLM"
},
"vocab_size": 50304,
"d_model": 1536,
"n_layers": 28,
"n_heads": 24,
"n_kv_heads": 6,
"max_len": 2048,
"rope_theta": 5000000.0,
"ffn_hidden": 4096,
"use_moe": true,
"moe_experts": 16,
"moe_top_k": 2,
"moe_shared": 2,
"moe_hidden": 1024,
"moe_every": 2,
"moe_bias_speed": 0.001,
"use_qk_norm": true,
"use_structural_bias": true,
"n_struct_rel": 4,
"use_fim": true,
"fim_rate": 0.5,
"fim_pre_id": 50269,
"fim_suf_id": 50270,
"fim_mid_id": 50271,
"tie_word_embeddings": true,
"torch_dtype": "bfloat16",
"bos_token_id": 50256,
"eos_token_id": 50256,
"pad_token_id": 50256,
"transformers_version": "4.44.0"
}