Sol-Lite-2 / config.json
j0no12's picture
Upload final Sol Lite 2 with Sol2ForCausalLM and banner
53d1439 verified
Raw History Blame Contribute Delete
620 Bytes
{
"vocab_size": 4096,
"width": 256,
"heads": 8,
"kv_heads": 4,
"blocks": 10,
"ffn_width": 1552,
"recurrent_start": 1,
"recurrent_blocks": 4,
"passes": 2,
"loop_conditioning": true,
"rope_theta": 20000.0,
"max_context": 2048,
"model_type": "sol2",
"architectures": [
"Sol2ForCausalLM"
],
"backend": "sdpa",
"torch_dtype": "float32",
"tie_word_embeddings": true,
"is_decoder": true,
"use_cache": false,
"eos_token_id": 3,
"pad_token_id": 3,
"auto_map": {
"AutoConfig": "configuration_sol2.Sol2Config",
"AutoModelForCausalLM": "modeling_sol2.Sol2ForCausalLM"
}
}