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"
}