korbip's picture
Add ComplexKDA 1.3B/100BT FineWeb-Edu model
92db682 verified
Raw History Blame Contribute Delete
1.23 kB
{
"architectures": [
"ComplexKDAForCausalLM"
],
"model_type": "complex_kda",
"auto_map": {
"AutoConfig": "configuration_complex_kda.ComplexKDAConfig",
"AutoModel": "modeling_complex_kda.ComplexKDAModel",
"AutoModelForCausalLM": "modeling_complex_kda.ComplexKDAForCausalLM"
},
"attn_mode": "chunk",
"hidden_size": 2048,
"num_hidden_layers": 24,
"num_heads": 16,
"head_dim": 128,
"intermediate_size": 5312,
"hidden_ratio": null,
"hidden_act": "swish",
"vocab_size": 32000,
"max_position_embeddings": 4096,
"tie_word_embeddings": false,
"norm_eps": 1e-06,
"conv_size": 4,
"use_short_conv": true,
"fuse_norm": true,
"fuse_swiglu": true,
"fuse_cross_entropy": true,
"initializer_range": 0.02,
"allow_neg_eigval": false,
"lower_bound": -5.0,
"expand_v": 1.0,
"gate": "sigmoid",
"gate_init_style": "shipped",
"output_gate": "lowrank",
"attn": {
"layers": [
3,
7,
11,
15,
19,
23
],
"num_heads": 16,
"num_kv_heads": 16,
"qkv_bias": false,
"qk_norm": false,
"output_gate": true,
"use_rope": false,
"rope_theta": 10000.0,
"window_size": null
},
"drop_silu": true,
"use_cache": true
}