RyanMullins commited on
Commit
cbf3b43
·
1 Parent(s): 3db648a

Gemma 4 E2B

Browse files
Files changed (4) hide show
  1. config.json +21 -6
  2. model.safetensors +2 -2
  3. processor_config.json +25 -1
  4. tokenizer.json +2 -2
config.json CHANGED
@@ -3,6 +3,9 @@
3
  "Gemma4ForConditionalGeneration"
4
  ],
5
  "audio_config": {
 
 
 
6
  "conf_attention_chunk_size": 12,
7
  "conf_attention_context_left": 13,
8
  "conf_attention_context_right": 0,
@@ -19,9 +22,22 @@
19
  "embedding_norm_eps": 1e-06,
20
  "gradient_clipping": 10000000000.0,
21
  "hidden_size": 1024,
 
 
 
 
22
  "input_feat_size": 128,
 
 
 
 
 
23
  "model_type": "gemma4_audio",
 
 
24
  "output_proj_dims": 1536,
 
 
25
  "rms_norm_eps": 1e-06,
26
  "sscp_conv_channel_size": [
27
  128,
@@ -56,10 +72,12 @@
56
  "streaming": false,
57
  "streaming_chunk_size": 51,
58
  "streaming_stride": 48,
 
59
  "vocab_offset": 262272,
60
  "vocab_size": 128
61
  },
62
- "audio_soft_tokens_per_image": 188,
 
63
  "audio_token_id": 258881,
64
  "boa_token_id": 256000,
65
  "boi_token_id": 255999,
@@ -78,11 +96,11 @@
78
  "attn_logit_softcapping": null,
79
  "bos_token_id": 2,
80
  "dtype": "bfloat16",
 
81
  "eos_token_id": 1,
82
  "expert_intermediate_size": null,
83
  "final_logit_softcapping": 30.0,
84
  "global_head_dim": 512,
85
- "global_partial_rotary_factor": 0.25,
86
  "head_dim": 256,
87
  "hidden_activation": "gelu_pytorch_tanh",
88
  "hidden_size": 1536,
@@ -135,17 +153,15 @@
135
  "num_key_value_heads": 1,
136
  "num_kv_shared_layers": 20,
137
  "pad_token_id": 0,
138
- "partial_rotary_factor": 1.0,
139
  "query_pre_attn_scalar": 256,
140
  "rms_norm_eps": 1e-06,
141
  "rope_parameters": {
142
  "full_attention": {
143
- "partial_rotary_factor": 1.0,
144
  "rope_theta": 1000000.0,
145
  "rope_type": "proportional"
146
  },
147
  "sliding_attention": {
148
- "partial_rotary_factor": 1.0,
149
  "rope_theta": 10000.0,
150
  "rope_type": "default"
151
  }
@@ -159,7 +175,6 @@
159
  "use_bidirectional_attention": null,
160
  "use_cache": true,
161
  "use_double_wide_mlp": true,
162
- "use_second_mlp_block": false,
163
  "vocab_size": 262144,
164
  "vocab_size_per_layer_input": 262144
165
  },
 
3
  "Gemma4ForConditionalGeneration"
4
  ],
5
  "audio_config": {
6
+ "_name_or_path": "",
7
+ "architectures": null,
8
+ "chunk_size_feed_forward": 0,
9
  "conf_attention_chunk_size": 12,
10
  "conf_attention_context_left": 13,
11
  "conf_attention_context_right": 0,
 
22
  "embedding_norm_eps": 1e-06,
23
  "gradient_clipping": 10000000000.0,
24
  "hidden_size": 1024,
25
+ "id2label": {
26
+ "0": "LABEL_0",
27
+ "1": "LABEL_1"
28
+ },
29
  "input_feat_size": 128,
30
+ "is_encoder_decoder": false,
31
+ "label2id": {
32
+ "LABEL_0": 0,
33
+ "LABEL_1": 1
34
+ },
35
  "model_type": "gemma4_audio",
36
+ "output_attentions": false,
37
+ "output_hidden_states": false,
38
  "output_proj_dims": 1536,
39
+ "problem_type": null,
40
+ "return_dict": true,
41
  "rms_norm_eps": 1e-06,
42
  "sscp_conv_channel_size": [
43
  128,
 
72
  "streaming": false,
73
  "streaming_chunk_size": 51,
74
  "streaming_stride": 48,
75
+ "use_clipped_linears": true,
76
  "vocab_offset": 262272,
77
  "vocab_size": 128
78
  },
79
+ "audio_ms_per_token": 40,
80
+ "audio_soft_tokens_per_image": 750,
81
  "audio_token_id": 258881,
82
  "boa_token_id": 256000,
83
  "boi_token_id": 255999,
 
96
  "attn_logit_softcapping": null,
97
  "bos_token_id": 2,
98
  "dtype": "bfloat16",
99
+ "enable_moe_block": false,
100
  "eos_token_id": 1,
101
  "expert_intermediate_size": null,
102
  "final_logit_softcapping": 30.0,
103
  "global_head_dim": 512,
 
104
  "head_dim": 256,
105
  "hidden_activation": "gelu_pytorch_tanh",
106
  "hidden_size": 1536,
 
153
  "num_key_value_heads": 1,
154
  "num_kv_shared_layers": 20,
155
  "pad_token_id": 0,
 
156
  "query_pre_attn_scalar": 256,
157
  "rms_norm_eps": 1e-06,
158
  "rope_parameters": {
159
  "full_attention": {
160
+ "partial_rotary_factor": 0.25,
161
  "rope_theta": 1000000.0,
162
  "rope_type": "proportional"
163
  },
164
  "sliding_attention": {
 
165
  "rope_theta": 10000.0,
166
  "rope_type": "default"
167
  }
 
175
  "use_bidirectional_attention": null,
176
  "use_cache": true,
177
  "use_double_wide_mlp": true,
 
178
  "vocab_size": 262144,
179
  "vocab_size_per_layer_input": 262144
180
  },
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c17cc59dd92ed94c5d01c2980107d67c06439d21924924d56987dc5ffaaa3f15
3
- size 10246559478
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e908419f2891b4cddaff87506a21aa22fce730e9dac81ef1ae59d85d412d8f6
3
+ size 10246627534
processor_config.json CHANGED
@@ -1,5 +1,29 @@
1
  {
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  "image_processor": {
 
3
  "do_convert_rgb": true,
4
  "do_normalize": false,
5
  "do_rescale": true,
@@ -9,7 +33,7 @@
9
  0.5,
10
  0.5
11
  ],
12
- "image_processor_type": "Gemma4ImageProcessor",
13
  "image_seq_length": 280,
14
  "image_std": [
15
  0.5,
 
1
  {
2
+ "audio_ms_per_token": 40,
3
+ "audio_seq_length": 750,
4
+ "feature_extractor": {
5
+ "dither": 0.0,
6
+ "feature_extractor_type": "Gemma4AudioFeatureExtractor",
7
+ "feature_size": 128,
8
+ "fft_length": 1024,
9
+ "fft_overdrive": true,
10
+ "frame_length": 320,
11
+ "hop_length": 160,
12
+ "input_scale_factor": 1.0,
13
+ "max_frequency": 8000.0,
14
+ "mel_floor": 0.001,
15
+ "min_frequency": 0.0,
16
+ "padding_side": "right",
17
+ "padding_value": 0.0,
18
+ "per_bin_mean": null,
19
+ "per_bin_stddev": null,
20
+ "preemphasis": 0.0,
21
+ "preemphasis_htk_flavor": true,
22
+ "return_attention_mask": true,
23
+ "sampling_rate": 16000
24
+ },
25
  "image_processor": {
26
+ "data_format": "channels_first",
27
  "do_convert_rgb": true,
28
  "do_normalize": false,
29
  "do_rescale": true,
 
33
  0.5,
34
  0.5
35
  ],
36
+ "image_processor_type": "Gemma4ImageProcessorFast",
37
  "image_seq_length": 280,
38
  "image_std": [
39
  0.5,
tokenizer.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3e9f1033df7d7df8240d1c80089baee77df1980b4131a8be7beb334ec8a48fa2
3
- size 32169763
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6f9cb1153f49bc7e0c148d0c1766017aed9512f45eb33afeeb71565d29c938b
3
+ size 32169319