| { |
| "architectures": [ |
| "KimiVLForConditionalGeneration" |
| ], |
| "attn_mechanism": "ragged_page_attention_v3", |
| "auto_map": { |
| "AutoConfig": "configuration_kimi_vl.KimiVLConfig", |
| "AutoModel": "modeling_kimi_vl.KimiVLForConditionalGeneration", |
| "AutoModelForCausalLM": "modeling_kimi_vl.KimiVLForConditionalGeneration" |
| }, |
| "backend": null, |
| "bits": null, |
| "blocksize_b": 1, |
| "blocksize_k": 128, |
| "blocksize_q": 128, |
| "decode_attn_mechanism": null, |
| "dtype": "bfloat16", |
| "easy_method": "train", |
| "fcm_max_ratio": 0.0, |
| "fcm_min_ratio": 0.0, |
| "flash_attention_backward_pass_impl": "triton", |
| "freq_max_position_embeddings": 40960, |
| "fsdp_is_ep_bound": true, |
| "gradient_checkpointing": "", |
| "gradient_checkpointing_targets": null, |
| "hardware_abstraction": true, |
| "ignore_index": -100, |
| "kv_cache_quantization_config": null, |
| "kv_cache_sharding_sequence_axis_name": "sp", |
| "mask_max_position_embeddings": 40960, |
| "media_placeholder_token_id": 163605, |
| "model_type": "kimi_vl", |
| "moe_force_xla_gmm": false, |
| "moe_method": "fused_moe", |
| "moe_tiling_size_batch": 4, |
| "moe_tiling_size_dim": 128, |
| "moe_tiling_size_seqlen": 128, |
| "operation_configs": null, |
| "pad_token_id": 0, |
| "pallas_k_block_size": 128, |
| "pallas_m_block_size": 128, |
| "pallas_n_block_size": 128, |
| "partition_axis": { |
| "attention_dim_axis": null, |
| "attention_kv_dim_axis": null, |
| "batch_axis": [ |
| "fsdp", |
| "dp" |
| ], |
| "bias_head_sequence_axis": null, |
| "bias_key_sequence_axis": null, |
| "data_parallel_axis": "dp", |
| "decode_attention_dim_axis": null, |
| "decode_attention_kv_dim_axis": null, |
| "decode_batch_axis": [ |
| "fsdp", |
| "dp" |
| ], |
| "decode_head_axis": "tp", |
| "decode_key_sequence_axis": "sp", |
| "decode_kv_head_axis": "tp", |
| "decode_query_sequence_axis": null, |
| "expert_axis": "ep", |
| "expert_gate_axis": null, |
| "expert_parallel_axis": "ep", |
| "fully_sharded_data_parallel_axis": "fsdp", |
| "head_axis": "tp", |
| "hidden_state_axis": "tp", |
| "key_sequence_axis": "sp", |
| "kv_head_axis": "tp", |
| "mlp_intermediate_axis": "tp", |
| "query_sequence_axis": "sp", |
| "sequence_axis": "sp", |
| "sequence_parallel_axis": "sp", |
| "tensor_parallel_axis": "tp", |
| "vocab_axis": "tp" |
| }, |
| "platform": null, |
| "precompute_masks": true, |
| "pretraining_tp": 1, |
| "quantization_config": null, |
| "scan_attention_layers": false, |
| "scan_mlp_chunk_size": 1024, |
| "scan_ring_attention": true, |
| "sequence_axis_name": "sp", |
| "sharding_axis_dims": [ |
| 1, |
| 1, |
| 1, |
| -1, |
| 1 |
| ], |
| "sharding_axis_names": [ |
| "dp", |
| "fsdp", |
| "ep", |
| "tp", |
| "sp" |
| ], |
| "sharding_dcn_axis_dims": null, |
| "sp_is_ep_bound": true, |
| "text_config": { |
| "architectures": [ |
| "KimiVLForConditionalGeneration" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "attn_mechanism": "ragged_page_attention_v3", |
| "auto_map": { |
| "AutoConfig": "configuration_kimi_vl.KimiVLConfig", |
| "AutoModel": "modeling_kimi_vl.KimiVLForConditionalGeneration", |
| "AutoModelForCausalLM": "modeling_kimi_vl.KimiVLForConditionalGeneration" |
| }, |
| "aux_loss_alpha": 0.001, |
| "backend": null, |
| "bits": null, |
| "blocksize_b": 1, |
| "blocksize_k": 128, |
| "blocksize_q": 128, |
| "bos_token_id": 163584, |
| "decode_attn_mechanism": null, |
| "dtype": "bfloat16", |
| "easy_method": "train", |
| "eos_token_id": 163585, |
| "ep_size": 1, |
| "fcm_max_ratio": 0.0, |
| "fcm_min_ratio": 0.0, |
| "first_k_dense_replace": 1, |
| "flash_attention_backward_pass_impl": "triton", |
| "freq_max_position_embeddings": 40960, |
| "fsdp_is_ep_bound": true, |
| "gradient_checkpointing": "", |
| "gradient_checkpointing_targets": null, |
| "hardware_abstraction": true, |
| "head_dim": 192, |
| "hidden_act": "silu", |
| "hidden_size": 2048, |
| "initializer_range": 0.02, |
| "intermediate_size": 11264, |
| "kv_cache_quantization_config": null, |
| "kv_cache_sharding_sequence_axis_name": "sp", |
| "kv_lora_rank": 512, |
| "layer_types": [ |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention", |
| "full_attention" |
| ], |
| "mask_max_position_embeddings": 40960, |
| "max_position_embeddings": 131072, |
| "model_type": "deepseek_v3", |
| "moe_force_xla_gmm": false, |
| "moe_intermediate_size": 1408, |
| "moe_layer_freq": 1, |
| "moe_method": "fused_moe", |
| "moe_tiling_size_batch": 4, |
| "moe_tiling_size_dim": 128, |
| "moe_tiling_size_seqlen": 128, |
| "n_group": 1, |
| "n_routed_experts": 64, |
| "n_shared_experts": 2, |
| "norm_topk_prob": true, |
| "num_attention_heads": 16, |
| "num_experts_per_tok": 6, |
| "num_hidden_layers": 27, |
| "num_key_value_heads": 16, |
| "num_nextn_predict_layers": 1, |
| "operation_configs": null, |
| "pad_token_id": 163839, |
| "pallas_k_block_size": 128, |
| "pallas_m_block_size": 128, |
| "pallas_n_block_size": 128, |
| "partition_axis": { |
| "attention_dim_axis": null, |
| "attention_kv_dim_axis": null, |
| "batch_axis": [ |
| "fsdp", |
| "dp" |
| ], |
| "bias_head_sequence_axis": null, |
| "bias_key_sequence_axis": null, |
| "data_parallel_axis": "dp", |
| "decode_attention_dim_axis": null, |
| "decode_attention_kv_dim_axis": null, |
| "decode_batch_axis": [ |
| "fsdp", |
| "dp" |
| ], |
| "decode_head_axis": "tp", |
| "decode_key_sequence_axis": "sp", |
| "decode_kv_head_axis": "tp", |
| "decode_query_sequence_axis": null, |
| "expert_axis": "ep", |
| "expert_gate_axis": null, |
| "expert_parallel_axis": "ep", |
| "fully_sharded_data_parallel_axis": "fsdp", |
| "head_axis": "tp", |
| "hidden_state_axis": "tp", |
| "key_sequence_axis": "sp", |
| "kv_head_axis": "tp", |
| "mlp_intermediate_axis": "tp", |
| "query_sequence_axis": "sp", |
| "sequence_axis": "sp", |
| "sequence_parallel_axis": "sp", |
| "tensor_parallel_axis": "tp", |
| "vocab_axis": "tp" |
| }, |
| "platform": null, |
| "precompute_masks": true, |
| "pretraining_tp": 1, |
| "q_lora_rank": null, |
| "qk_nope_head_dim": 128, |
| "qk_rope_head_dim": 64, |
| "quantization_config": null, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": null, |
| "rope_theta": 800000.0, |
| "routed_scaling_factor": 2.446, |
| "scan_attention_layers": false, |
| "scan_mlp_chunk_size": 1024, |
| "scan_ring_attention": true, |
| "scoring_func": "sigmoid", |
| "seq_aux": true, |
| "sequence_axis_name": "sp", |
| "sharding_axis_dims": [ |
| 1, |
| 1, |
| 1, |
| -1, |
| 1 |
| ], |
| "sharding_axis_names": [ |
| "dp", |
| "fsdp", |
| "ep", |
| "tp", |
| "sp" |
| ], |
| "sharding_dcn_axis_dims": null, |
| "sp_is_ep_bound": true, |
| "topk_group": 1, |
| "topk_method": "noaux_tc", |
| "use_cache": true, |
| "use_expert_tensor_mode": false, |
| "use_ring_of_experts": false, |
| "use_scan_mlp": false, |
| "use_sharded_kv_caching": false, |
| "use_sharding_constraint": false, |
| "v_head_dim": 128, |
| "vocab_size": 163840 |
| }, |
| "tie_word_embeddings": false, |
| "transformers_version": "4.57.3", |
| "use_expert_tensor_mode": false, |
| "use_ring_of_experts": false, |
| "use_scan_mlp": false, |
| "use_sharded_kv_caching": false, |
| "use_sharding_constraint": false, |
| "vision_config": { |
| "architectures": [ |
| "KimiVLForConditionalGeneration" |
| ], |
| "attn_mechanism": "ragged_page_attention_v3", |
| "auto_map": { |
| "AutoConfig": "configuration_kimi_vl.KimiVLConfig", |
| "AutoModel": "modeling_kimi_vl.KimiVLForConditionalGeneration", |
| "AutoModelForCausalLM": "modeling_kimi_vl.KimiVLForConditionalGeneration" |
| }, |
| "backend": null, |
| "bits": null, |
| "blocksize_b": 1, |
| "blocksize_k": 128, |
| "blocksize_q": 128, |
| "decode_attn_mechanism": null, |
| "dtype": "bfloat16", |
| "easy_method": "train", |
| "fcm_max_ratio": 0.0, |
| "fcm_min_ratio": 0.0, |
| "flash_attention_backward_pass_impl": "triton", |
| "freq_max_position_embeddings": 40960, |
| "fsdp_is_ep_bound": true, |
| "gradient_checkpointing": "", |
| "gradient_checkpointing_targets": null, |
| "hardware_abstraction": true, |
| "hidden_size": 1152, |
| "init_pos_emb_height": 64, |
| "init_pos_emb_width": 64, |
| "intermediate_size": 4304, |
| "kv_cache_quantization_config": null, |
| "kv_cache_sharding_sequence_axis_name": "sp", |
| "mask_max_position_embeddings": 40960, |
| "merge_kernel_size": [ |
| 2, |
| 2 |
| ], |
| "model_type": "moonvit", |
| "moe_force_xla_gmm": false, |
| "moe_method": "fused_moe", |
| "moe_tiling_size_batch": 4, |
| "moe_tiling_size_dim": 128, |
| "moe_tiling_size_seqlen": 128, |
| "num_attention_heads": 16, |
| "num_hidden_layers": 27, |
| "operation_configs": null, |
| "pallas_k_block_size": 128, |
| "pallas_m_block_size": 128, |
| "pallas_n_block_size": 128, |
| "partition_axis": { |
| "attention_dim_axis": null, |
| "attention_kv_dim_axis": null, |
| "batch_axis": [ |
| "fsdp", |
| "dp" |
| ], |
| "bias_head_sequence_axis": null, |
| "bias_key_sequence_axis": null, |
| "data_parallel_axis": "dp", |
| "decode_attention_dim_axis": null, |
| "decode_attention_kv_dim_axis": null, |
| "decode_batch_axis": [ |
| "fsdp", |
| "dp" |
| ], |
| "decode_head_axis": "tp", |
| "decode_key_sequence_axis": "sp", |
| "decode_kv_head_axis": "tp", |
| "decode_query_sequence_axis": null, |
| "expert_axis": "ep", |
| "expert_gate_axis": null, |
| "expert_parallel_axis": "ep", |
| "fully_sharded_data_parallel_axis": "fsdp", |
| "head_axis": "tp", |
| "hidden_state_axis": "tp", |
| "key_sequence_axis": "sp", |
| "kv_head_axis": "tp", |
| "mlp_intermediate_axis": "tp", |
| "query_sequence_axis": "sp", |
| "sequence_axis": "sp", |
| "sequence_parallel_axis": "sp", |
| "tensor_parallel_axis": "tp", |
| "vocab_axis": "tp" |
| }, |
| "patch_size": 14, |
| "platform": null, |
| "precompute_masks": true, |
| "pretraining_tp": 1, |
| "quantization_config": null, |
| "scan_attention_layers": false, |
| "scan_mlp_chunk_size": 1024, |
| "scan_ring_attention": true, |
| "sequence_axis_name": "sp", |
| "sharding_axis_dims": [ |
| 1, |
| 1, |
| 1, |
| -1, |
| 1 |
| ], |
| "sharding_axis_names": [ |
| "dp", |
| "fsdp", |
| "ep", |
| "tp", |
| "sp" |
| ], |
| "sharding_dcn_axis_dims": null, |
| "sp_is_ep_bound": true, |
| "use_expert_tensor_mode": false, |
| "use_ring_of_experts": false, |
| "use_scan_mlp": false, |
| "use_sharded_kv_caching": false, |
| "use_sharding_constraint": false, |
| "vocab_size": 163840 |
| }, |
| "vocab_size": 163840 |
| } |
|
|
