hf-transformers-bot's picture
Update tiny models for InklingForCausalLM
4d82a3b verified
Raw
History Blame Contribute Delete
1.35 kB
{
"architectures": [
"InklingForCausalLM"
],
"attention_dropout": 0.0,
"bos_token_id": 1,
"chain_hidden_post_norm": false,
"conv_kernel_size": 4,
"d_rel": 16,
"dtype": "float32",
"eos_token_id": 2,
"head_dim": 16,
"hidden_act": "gelu",
"hidden_size": 32,
"initializer_range": 0.02,
"intermediate_size": 32,
"is_decoder": true,
"layer_types": [
"hybrid_sliding",
"hybrid"
],
"local_layer_ids": null,
"log_scaling_alpha": 0.1,
"log_scaling_n_floor": null,
"logits_mup_width_multiplier": 24.0,
"max_position_embeddings": 512,
"mlp_layer_types": [
"dense",
"sparse"
],
"model_type": "inkling_text",
"moe_intermediate_size": 16,
"mtp_hidden_states_first": true,
"mtp_local_layer_ids": null,
"n_routed_experts": 256,
"n_shared_experts": 2,
"num_attention_heads": 2,
"num_experts_per_tok": 2,
"num_hidden_layers": 2,
"num_key_value_heads": 2,
"num_mtp_layers": null,
"number_of_conv_states": 4,
"pad_token_id": 0,
"rel_extent": 1024,
"rms_norm_eps": 1e-06,
"rms_norm_eps_moe_gate": 1e-06,
"route_scale": 8.0,
"shared_expert_sink": true,
"sliding_window_size": 512,
"swa_head_dim": 16,
"swa_num_attention_heads": 2,
"swa_num_key_value_heads": 2,
"transformers_version": "5.16.0.dev0",
"unpadded_vocab_size": null,
"vocab_size": 200058
}