Instructions to use google/gemma-4-E2B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use google/gemma-4-E2B with Transformers:
# Load model directly from transformers import AutoProcessor, AutoModelForMultimodalLM processor = AutoProcessor.from_pretrained("google/gemma-4-E2B") model = AutoModelForMultimodalLM.from_pretrained("google/gemma-4-E2B", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit ·
cbf3b43
1
Parent(s): 3db648a
Gemma 4 E2B
Browse files- config.json +21 -6
- model.safetensors +2 -2
- processor_config.json +25 -1
- 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 |
-
"
|
|
|
|
| 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":
|
| 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:
|
| 3 |
-
size
|
|
|
|
| 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": "
|
| 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:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b6f9cb1153f49bc7e0c148d0c1766017aed9512f45eb33afeeb71565d29c938b
|
| 3 |
+
size 32169319
|