hf-transformers-bot's picture
Update tiny models for KimiLinearForCausalLM
75adf85 verified
Raw History Blame Contribute Delete
1.21 kB
{
"architectures": [
"KimiLinearForCausalLM"
],
"attention_bias": false,
"attention_dropout": 0.0,
"bos_token_id": 163584,
"dtype": "float32",
"eos_token_id": 163585,
"head_dim": 16,
"hidden_act": "silu",
"hidden_size": 32,
"initializer_range": 0.02,
"intermediate_size": 32,
"is_decoder": true,
"kv_lora_rank": 16,
"layer_types": [
"linear_attention",
"full_attention"
],
"linear_conv_kernel_dim": 2,
"linear_head_dim": 16,
"linear_num_heads": 4,
"mlp_layer_types": [
"dense",
"sparse"
],
"model_max_length": 512,
"model_type": "kimi_linear",
"moe_intermediate_size": 16,
"moe_renormalize": true,
"num_attention_heads": 2,
"num_expert_group": 1,
"num_experts": 4,
"num_experts_per_token": 2,
"num_hidden_layers": 2,
"num_key_value_heads": 2,
"num_shared_experts": 1,
"pad_token_id": 163839,
"pretraining_tp": 1,
"q_lora_rank": null,
"qk_head_dim": 48,
"qk_nope_head_dim": 32,
"qk_rope_head_dim": 16,
"rms_norm_eps": 1e-05,
"routed_scaling_factor": 2.446,
"tie_word_embeddings": false,
"topk_group": 1,
"transformers_version": "5.19.0.dev0",
"use_cache": true,
"v_head_dim": 32,
"vocab_size": 163840
}