CMA-1M-Mini / config.json
User01110's picture
CMA-1M Mini release
564ee09
Raw
History Blame Contribute Delete
679 Bytes
{
"architectures": [
"CMAForCausalLM"
],
"auto_map": {
"AutoConfig": "modeling_cma.CMAConfig",
"AutoModelForCausalLM": "modeling_cma.CMAForCausalLM"
},
"model_type": "cma",
"vocab_size": 260,
"seq_len": 2048,
"max_position_embeddings": 2048,
"n_positions": 2048,
"n_ctx": 2048,
"d_model": 128,
"n_layers": 6,
"n_heads": 4,
"n_kv_heads": 2,
"chunk": 16,
"cma_heads": 2,
"expand": 2,
"cma_identity_prob": 0.9,
"bos_token_id": 1,
"eos_token_id": 2,
"pad_token_id": 0,
"tie_word_embeddings": true,
"safe_serialization": true,
"minimum_torch_version": "2.5",
"torch_dtype": "bfloat16",
"transformers_version": "5.13.0"
}