{ "model_type": "agnes", "architectures": [ "AgnesForConditionalGeneration" ], "auto_map": { "AutoConfig": "configuration_agnes.AgnesConfig", "AutoModel": "modeling_agnes.AgnesModel", "AutoModelForCausalLM": "modeling_agnes.AgnesForConditionalGeneration", "AutoModelForImageTextToText": "modeling_agnes.AgnesForConditionalGeneration" }, "text_config": { "model_type": "agnes_text", "hidden_size": 5120, "num_hidden_layers": 72, "intermediate_size": 17408, "parallel_ffn_intermediate_size": 2048, "vocab_size": 248320, "layer_types": [ "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_delta_attention", "agnes_global_attention" ], "global_attention_interval": 4, "num_attention_heads": 24, "num_key_value_heads": 4, "head_dim": 256, "attention_bias": false, "attention_dropout": 0.0, "attn_output_gate": true, "partial_rotary_factor": 0.25, "linear_num_key_heads": 16, "linear_num_value_heads": 48, "linear_key_head_dim": 128, "linear_value_head_dim": 128, "linear_conv_kernel_dim": 4, "output_gate_type": "swish", "mamba_ssm_dtype": "float32", "hidden_act": "silu", "rms_norm_eps": 1e-06, "initializer_range": 0.02, "rope_parameters": { "mrope_interleaved": true, "mrope_section": [ 11, 11, 10 ], "partial_rotary_factor": 0.25, "rope_theta": 10000000, "rope_type": "default" }, "max_position_embeddings": 262144, "mtp_num_hidden_layers": 1, "mtp_use_dedicated_embeddings": false, "bos_token_id": 248044, "eos_token_id": 248044, "pad_token_id": null, "tie_word_embeddings": false, "use_cache": true, "dtype": "bfloat16" }, "vision_config": { "model_type": "agnes_vision", "depth": 27, "hidden_size": 1152, "intermediate_size": 4304, "num_heads": 16, "out_hidden_size": 5120, "patch_size": 16, "spatial_merge_size": 2, "temporal_patch_size": 2, "in_channels": 3, "num_position_embeddings": 2304, "deepstack_visual_indexes": [], "hidden_act": "gelu_pytorch_tanh", "initializer_range": 0.02 }, "image_token_id": 248056, "video_token_id": 248057, "vision_start_token_id": 248053, "vision_end_token_id": 248054, "tie_word_embeddings": false }