File size: 4,798 Bytes
db3f57f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 | {
"model_name": "models--google--gemma-4-E2B-it",
"model_type": "vlm-gemma4",
"vm_cfg": {
"model_type": "gemma4_vision",
"arch": "gemma4_vision",
"image_size": [
480,
480
],
"patch_size": 16,
"cls_embed": false,
"hidden_size": 768,
"intermediate_size": 3072,
"num_attention_heads": 12,
"num_hidden_layers": 16,
"hidden_act": "gelu_pytorch_tanh",
"layer_norm_eps": 1e-06,
"spatial_merge_size": 3,
"temporal_patch_size": 1,
"window_size": 0,
"num_position_embeddings": 0,
"fullatt_block_indexes": [],
"deepstack_visual_indexes": [],
"rope_theta": 100.0
},
"mm_cfg": {
"num_layers": 2,
"hidden_act": "gelu",
"mm_tokens_per_image": 100,
"proj_dim": 1536,
"downsample_factor": 1
},
"lm_cfg": {
"model_type": "gemma4_text",
"data_type": "bfloat16",
"arch": "gemma",
"token_cfg": {
"vocab_size": 262144
},
"rope_cfg": {
"rope_theta": 1000000.0,
"rope_local_base_freq": 10000.0,
"rope_dimension_count": 128,
"sliding_rope_dimension_count": 256,
"rope_scaling": {
"factor": 1.0,
"low_freq_factor": 0,
"high_freq_factor": 0,
"original_max_position_embeddings": 0,
"long_factor": null,
"short_factor": null,
"rope_type": "proportional",
"mrope_section": null,
"mrope_interleaved": false
}
},
"attn_cfg": {
"num_attention_heads": 8,
"num_key_value_heads": 1,
"head_dim": 512,
"sliding_head_dim": 256,
"swa_enable": true,
"sliding_window": 512,
"attention_bias": false,
"attention_dropout": 0.0,
"query_pre_attn_scalar": 0
},
"mlp_cfg": {
"intermediate_size": 6144,
"act": "gelu_pytorch_tanh",
"num_layers": 3,
"mlp_bias": false
},
"hidden_size": 1536,
"num_hidden_layers": 35,
"max_position_embeddings": 2048,
"rms_norm_eps": 1e-06,
"rms_norm_unit_offset": false,
"layer_types": [
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention"
],
"hidden_size_per_layer_input": 256,
"num_kv_shared_layers": 20,
"use_double_wide_mlp": true,
"attn_logit_softcapping": null,
"final_logit_softcapping": 30.0,
"lm_head_num_splits": 4,
"lm_head_split_dim": 65536,
"conv_L_cache": 3,
"conv_bias": false,
"lora_cfg": null,
"draft_cfg": null
},
"pipeline_cfg": {
"system_prompt": null,
"chat_template": null,
"max_num_tokens": 2048,
"input_token_group_size": 128,
"input_token_group_offsets": [
0,
128,
256,
384,
512,
640,
768,
896,
1024,
1152,
1280,
1408,
1536,
1664,
1792,
1920
],
"future_token_mask_size": 128,
"return_logits": false,
"use_strided_kv_cache": false,
"enable_filter_sharing": false,
"quantize_embeddings": false,
"split_mlp": true
},
"language_model_name": "models--google--gemma-4-E2B-it_language",
"vision_model_name": "models--google--gemma-4-E2B-it_vision"
}
|