jerryzh168 commited on
Commit
3d604aa
·
verified ·
1 Parent(s): 7421b93

Upload Gemma3ForConditionalGeneration

Browse files
config.json CHANGED
@@ -3,6 +3,7 @@
3
  "Gemma3ForConditionalGeneration"
4
  ],
5
  "boi_token_index": 255999,
 
6
  "eoi_token_index": 256000,
7
  "eos_token_id": [
8
  1,
@@ -13,12 +14,15 @@
13
  "mm_tokens_per_image": 256,
14
  "model_type": "gemma3",
15
  "quantization_config": {
 
16
  "modules_to_not_convert": null,
17
  "quant_method": "torchao",
18
  "quant_type": {
19
  "default": {
20
  "_data": {
21
  "group_size": 128,
 
 
22
  "layout": {
23
  "_data": {
24
  "inner_k_tiles": 8
@@ -26,10 +30,6 @@
26
  "_type": "TensorCoreTiledLayout",
27
  "_version": 1
28
  },
29
- "packing_format": {
30
- "_data": "PLAIN",
31
- "_type": "PackingFormat"
32
- },
33
  "preserve_zero": null,
34
  "set_inductor_config": true,
35
  "use_hqq": false,
@@ -42,19 +42,71 @@
42
  "_version": 2
43
  }
44
  },
45
- "quant_type_kwargs": {}
 
46
  },
47
  "text_config": {
 
48
  "attention_bias": false,
49
  "attention_dropout": 0.0,
50
  "attn_logit_softcapping": null,
51
- "cache_implementation": "hybrid",
52
  "final_logit_softcapping": null,
53
  "head_dim": 256,
54
  "hidden_activation": "gelu_pytorch_tanh",
55
  "hidden_size": 3840,
56
  "initializer_range": 0.02,
57
  "intermediate_size": 15360,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
58
  "max_position_embeddings": 131072,
59
  "model_type": "gemma3_text",
60
  "num_attention_heads": 16,
@@ -69,15 +121,13 @@
69
  },
70
  "rope_theta": 1000000.0,
71
  "sliding_window": 1024,
72
- "sliding_window_pattern": 6,
73
- "torch_dtype": "bfloat16",
74
  "use_cache": true,
75
  "vocab_size": 262208
76
  },
77
- "torch_dtype": "bfloat16",
78
- "transformers_version": "4.52.0.dev0",
79
  "vision_config": {
80
  "attention_dropout": 0.0,
 
81
  "hidden_act": "gelu_pytorch_tanh",
82
  "hidden_size": 1152,
83
  "image_size": 896,
@@ -88,7 +138,6 @@
88
  "num_channels": 3,
89
  "num_hidden_layers": 27,
90
  "patch_size": 14,
91
- "torch_dtype": "bfloat16",
92
  "vision_use_head": false
93
  }
94
  }
 
3
  "Gemma3ForConditionalGeneration"
4
  ],
5
  "boi_token_index": 255999,
6
+ "dtype": "bfloat16",
7
  "eoi_token_index": 256000,
8
  "eos_token_id": [
9
  1,
 
14
  "mm_tokens_per_image": 256,
15
  "model_type": "gemma3",
16
  "quantization_config": {
17
+ "include_input_output_embeddings": false,
18
  "modules_to_not_convert": null,
19
  "quant_method": "torchao",
20
  "quant_type": {
21
  "default": {
22
  "_data": {
23
  "group_size": 128,
24
+ "int4_choose_qparams_algorithm": "hqq",
25
+ "int4_packing_format": "tile_packed_to_4d",
26
  "layout": {
27
  "_data": {
28
  "inner_k_tiles": 8
 
30
  "_type": "TensorCoreTiledLayout",
31
  "_version": 1
32
  },
 
 
 
 
33
  "preserve_zero": null,
34
  "set_inductor_config": true,
35
  "use_hqq": false,
 
42
  "_version": 2
43
  }
44
  },
45
+ "quant_type_kwargs": {},
46
+ "untie_embedding_weights": false
47
  },
48
  "text_config": {
49
+ "_sliding_window_pattern": 6,
50
  "attention_bias": false,
51
  "attention_dropout": 0.0,
52
  "attn_logit_softcapping": null,
53
+ "dtype": "bfloat16",
54
  "final_logit_softcapping": null,
55
  "head_dim": 256,
56
  "hidden_activation": "gelu_pytorch_tanh",
57
  "hidden_size": 3840,
58
  "initializer_range": 0.02,
59
  "intermediate_size": 15360,
60
+ "layer_types": [
61
+ "sliding_attention",
62
+ "sliding_attention",
63
+ "sliding_attention",
64
+ "sliding_attention",
65
+ "sliding_attention",
66
+ "full_attention",
67
+ "sliding_attention",
68
+ "sliding_attention",
69
+ "sliding_attention",
70
+ "sliding_attention",
71
+ "sliding_attention",
72
+ "full_attention",
73
+ "sliding_attention",
74
+ "sliding_attention",
75
+ "sliding_attention",
76
+ "sliding_attention",
77
+ "sliding_attention",
78
+ "full_attention",
79
+ "sliding_attention",
80
+ "sliding_attention",
81
+ "sliding_attention",
82
+ "sliding_attention",
83
+ "sliding_attention",
84
+ "full_attention",
85
+ "sliding_attention",
86
+ "sliding_attention",
87
+ "sliding_attention",
88
+ "sliding_attention",
89
+ "sliding_attention",
90
+ "full_attention",
91
+ "sliding_attention",
92
+ "sliding_attention",
93
+ "sliding_attention",
94
+ "sliding_attention",
95
+ "sliding_attention",
96
+ "full_attention",
97
+ "sliding_attention",
98
+ "sliding_attention",
99
+ "sliding_attention",
100
+ "sliding_attention",
101
+ "sliding_attention",
102
+ "full_attention",
103
+ "sliding_attention",
104
+ "sliding_attention",
105
+ "sliding_attention",
106
+ "sliding_attention",
107
+ "sliding_attention",
108
+ "full_attention"
109
+ ],
110
  "max_position_embeddings": 131072,
111
  "model_type": "gemma3_text",
112
  "num_attention_heads": 16,
 
121
  },
122
  "rope_theta": 1000000.0,
123
  "sliding_window": 1024,
 
 
124
  "use_cache": true,
125
  "vocab_size": 262208
126
  },
127
+ "transformers_version": "4.56.1",
 
128
  "vision_config": {
129
  "attention_dropout": 0.0,
130
+ "dtype": "bfloat16",
131
  "hidden_act": "gelu_pytorch_tanh",
132
  "hidden_size": 1152,
133
  "image_size": 896,
 
138
  "num_channels": 3,
139
  "num_hidden_layers": 27,
140
  "patch_size": 14,
 
141
  "vision_use_head": false
142
  }
143
  }
generation_config.json CHANGED
@@ -9,5 +9,5 @@
9
  "pad_token_id": 0,
10
  "top_k": 64,
11
  "top_p": 0.95,
12
- "transformers_version": "4.52.0.dev0"
13
  }
 
9
  "pad_token_id": 0,
10
  "top_k": 64,
11
  "top_p": 0.95,
12
+ "transformers_version": "4.56.1"
13
  }
pytorch_model-00001-of-00002.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:538b990ced48d6dca85707e82960f5d82522e8fa1dee7ce94c6bf6d36f8026c4
3
- size 4975899385
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44f3a62d7487c8181bb87268e2bfd6ca40e0f9a7973b6f7df65d946ec0e39249
3
+ size 4987219422
pytorch_model-00002-of-00002.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:baa4d2f81b76200df8e943c0e4eab95336b5782b36c09e8a731e54191ff2a0dd
3
- size 3190968481
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:62ac4c77cfe8753a7375fc0fd69deea6326dcab87451b42455663741f32a381a
3
+ size 3548065336
pytorch_model.bin.index.json CHANGED
@@ -1,6 +1,7 @@
1
  {
2
  "metadata": {
3
- "total_size": 8166129312
 
4
  },
5
  "weight_map": {
6
  "language_model.lm_head.weight": "pytorch_model-00001-of-00002.bin",
@@ -148,13 +149,13 @@
148
  "language_model.model.layers.18.self_attn.q_norm.weight": "pytorch_model-00001-of-00002.bin",
149
  "language_model.model.layers.18.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
150
  "language_model.model.layers.18.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
151
- "language_model.model.layers.19.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
152
- "language_model.model.layers.19.mlp.down_proj.weight": "pytorch_model-00001-of-00002.bin",
153
  "language_model.model.layers.19.mlp.gate_proj.weight": "pytorch_model-00001-of-00002.bin",
154
- "language_model.model.layers.19.mlp.up_proj.weight": "pytorch_model-00001-of-00002.bin",
155
- "language_model.model.layers.19.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
156
- "language_model.model.layers.19.post_feedforward_layernorm.weight": "pytorch_model-00001-of-00002.bin",
157
- "language_model.model.layers.19.pre_feedforward_layernorm.weight": "pytorch_model-00001-of-00002.bin",
158
  "language_model.model.layers.19.self_attn.k_norm.weight": "pytorch_model-00001-of-00002.bin",
159
  "language_model.model.layers.19.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
160
  "language_model.model.layers.19.self_attn.o_proj.weight": "pytorch_model-00001-of-00002.bin",
@@ -174,19 +175,19 @@
174
  "language_model.model.layers.2.self_attn.q_norm.weight": "pytorch_model-00001-of-00002.bin",
175
  "language_model.model.layers.2.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
176
  "language_model.model.layers.2.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
177
- "language_model.model.layers.20.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
178
- "language_model.model.layers.20.mlp.down_proj.weight": "pytorch_model-00001-of-00002.bin",
179
- "language_model.model.layers.20.mlp.gate_proj.weight": "pytorch_model-00001-of-00002.bin",
180
- "language_model.model.layers.20.mlp.up_proj.weight": "pytorch_model-00001-of-00002.bin",
181
- "language_model.model.layers.20.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
182
- "language_model.model.layers.20.post_feedforward_layernorm.weight": "pytorch_model-00001-of-00002.bin",
183
- "language_model.model.layers.20.pre_feedforward_layernorm.weight": "pytorch_model-00001-of-00002.bin",
184
- "language_model.model.layers.20.self_attn.k_norm.weight": "pytorch_model-00001-of-00002.bin",
185
- "language_model.model.layers.20.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
186
- "language_model.model.layers.20.self_attn.o_proj.weight": "pytorch_model-00001-of-00002.bin",
187
- "language_model.model.layers.20.self_attn.q_norm.weight": "pytorch_model-00001-of-00002.bin",
188
- "language_model.model.layers.20.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
189
- "language_model.model.layers.20.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
190
  "language_model.model.layers.21.input_layernorm.weight": "pytorch_model-00002-of-00002.bin",
191
  "language_model.model.layers.21.mlp.down_proj.weight": "pytorch_model-00002-of-00002.bin",
192
  "language_model.model.layers.21.mlp.gate_proj.weight": "pytorch_model-00002-of-00002.bin",
@@ -194,12 +195,12 @@
194
  "language_model.model.layers.21.post_attention_layernorm.weight": "pytorch_model-00002-of-00002.bin",
195
  "language_model.model.layers.21.post_feedforward_layernorm.weight": "pytorch_model-00002-of-00002.bin",
196
  "language_model.model.layers.21.pre_feedforward_layernorm.weight": "pytorch_model-00002-of-00002.bin",
197
- "language_model.model.layers.21.self_attn.k_norm.weight": "pytorch_model-00001-of-00002.bin",
198
- "language_model.model.layers.21.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
199
- "language_model.model.layers.21.self_attn.o_proj.weight": "pytorch_model-00001-of-00002.bin",
200
- "language_model.model.layers.21.self_attn.q_norm.weight": "pytorch_model-00001-of-00002.bin",
201
- "language_model.model.layers.21.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
202
- "language_model.model.layers.21.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
203
  "language_model.model.layers.22.input_layernorm.weight": "pytorch_model-00002-of-00002.bin",
204
  "language_model.model.layers.22.mlp.down_proj.weight": "pytorch_model-00002-of-00002.bin",
205
  "language_model.model.layers.22.mlp.gate_proj.weight": "pytorch_model-00002-of-00002.bin",
 
1
  {
2
  "metadata": {
3
+ "total_parameters": 12187325040,
4
+ "total_size": 8534683872
5
  },
6
  "weight_map": {
7
  "language_model.lm_head.weight": "pytorch_model-00001-of-00002.bin",
 
149
  "language_model.model.layers.18.self_attn.q_norm.weight": "pytorch_model-00001-of-00002.bin",
150
  "language_model.model.layers.18.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
151
  "language_model.model.layers.18.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
152
+ "language_model.model.layers.19.input_layernorm.weight": "pytorch_model-00002-of-00002.bin",
153
+ "language_model.model.layers.19.mlp.down_proj.weight": "pytorch_model-00002-of-00002.bin",
154
  "language_model.model.layers.19.mlp.gate_proj.weight": "pytorch_model-00001-of-00002.bin",
155
+ "language_model.model.layers.19.mlp.up_proj.weight": "pytorch_model-00002-of-00002.bin",
156
+ "language_model.model.layers.19.post_attention_layernorm.weight": "pytorch_model-00002-of-00002.bin",
157
+ "language_model.model.layers.19.post_feedforward_layernorm.weight": "pytorch_model-00002-of-00002.bin",
158
+ "language_model.model.layers.19.pre_feedforward_layernorm.weight": "pytorch_model-00002-of-00002.bin",
159
  "language_model.model.layers.19.self_attn.k_norm.weight": "pytorch_model-00001-of-00002.bin",
160
  "language_model.model.layers.19.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
161
  "language_model.model.layers.19.self_attn.o_proj.weight": "pytorch_model-00001-of-00002.bin",
 
175
  "language_model.model.layers.2.self_attn.q_norm.weight": "pytorch_model-00001-of-00002.bin",
176
  "language_model.model.layers.2.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
177
  "language_model.model.layers.2.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
178
+ "language_model.model.layers.20.input_layernorm.weight": "pytorch_model-00002-of-00002.bin",
179
+ "language_model.model.layers.20.mlp.down_proj.weight": "pytorch_model-00002-of-00002.bin",
180
+ "language_model.model.layers.20.mlp.gate_proj.weight": "pytorch_model-00002-of-00002.bin",
181
+ "language_model.model.layers.20.mlp.up_proj.weight": "pytorch_model-00002-of-00002.bin",
182
+ "language_model.model.layers.20.post_attention_layernorm.weight": "pytorch_model-00002-of-00002.bin",
183
+ "language_model.model.layers.20.post_feedforward_layernorm.weight": "pytorch_model-00002-of-00002.bin",
184
+ "language_model.model.layers.20.pre_feedforward_layernorm.weight": "pytorch_model-00002-of-00002.bin",
185
+ "language_model.model.layers.20.self_attn.k_norm.weight": "pytorch_model-00002-of-00002.bin",
186
+ "language_model.model.layers.20.self_attn.k_proj.weight": "pytorch_model-00002-of-00002.bin",
187
+ "language_model.model.layers.20.self_attn.o_proj.weight": "pytorch_model-00002-of-00002.bin",
188
+ "language_model.model.layers.20.self_attn.q_norm.weight": "pytorch_model-00002-of-00002.bin",
189
+ "language_model.model.layers.20.self_attn.q_proj.weight": "pytorch_model-00002-of-00002.bin",
190
+ "language_model.model.layers.20.self_attn.v_proj.weight": "pytorch_model-00002-of-00002.bin",
191
  "language_model.model.layers.21.input_layernorm.weight": "pytorch_model-00002-of-00002.bin",
192
  "language_model.model.layers.21.mlp.down_proj.weight": "pytorch_model-00002-of-00002.bin",
193
  "language_model.model.layers.21.mlp.gate_proj.weight": "pytorch_model-00002-of-00002.bin",
 
195
  "language_model.model.layers.21.post_attention_layernorm.weight": "pytorch_model-00002-of-00002.bin",
196
  "language_model.model.layers.21.post_feedforward_layernorm.weight": "pytorch_model-00002-of-00002.bin",
197
  "language_model.model.layers.21.pre_feedforward_layernorm.weight": "pytorch_model-00002-of-00002.bin",
198
+ "language_model.model.layers.21.self_attn.k_norm.weight": "pytorch_model-00002-of-00002.bin",
199
+ "language_model.model.layers.21.self_attn.k_proj.weight": "pytorch_model-00002-of-00002.bin",
200
+ "language_model.model.layers.21.self_attn.o_proj.weight": "pytorch_model-00002-of-00002.bin",
201
+ "language_model.model.layers.21.self_attn.q_norm.weight": "pytorch_model-00002-of-00002.bin",
202
+ "language_model.model.layers.21.self_attn.q_proj.weight": "pytorch_model-00002-of-00002.bin",
203
+ "language_model.model.layers.21.self_attn.v_proj.weight": "pytorch_model-00002-of-00002.bin",
204
  "language_model.model.layers.22.input_layernorm.weight": "pytorch_model-00002-of-00002.bin",
205
  "language_model.model.layers.22.mlp.down_proj.weight": "pytorch_model-00002-of-00002.bin",
206
  "language_model.model.layers.22.mlp.gate_proj.weight": "pytorch_model-00002-of-00002.bin",