embeddinggemma-2-mic / build.json
tarekziade's picture
tarekziade HF Staff
Publish MIC usage, slim fixture and reviewed Transformers expectations
839a2b1 verified
Raw History Blame Contribute Delete
7.39 kB
{
"fixture": "huggingface-mic/embeddinggemma-2-mic",
"key": "9f46227c73ff97757360fe4087b3d34af70337be7ec361fa5a6b042c6ff3e365",
"key_fields": {
"architectures": [
"EmbeddingGemma2Model"
],
"output_hidden_states": false,
"return_dict": true,
"chunk_size_feed_forward": 0,
"is_encoder_decoder": false,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1"
},
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1
},
"problem_type": null,
"text_config": {
"architectures": null,
"output_hidden_states": false,
"return_dict": true,
"dtype": "bfloat16",
"chunk_size_feed_forward": 0,
"is_encoder_decoder": false,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1"
},
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1
},
"problem_type": null,
"vocab_size": 262144,
"hidden_size": 64,
"intermediate_size": 128,
"num_hidden_layers": 2,
"num_attention_heads": 4,
"num_key_value_heads": 2,
"head_dim": 16,
"hidden_activation": "gelu_pytorch_tanh",
"max_position_embeddings": 262144,
"initializer_range": 0.02,
"rms_norm_eps": 1e-06,
"pad_token_id": 0,
"eos_token_id": 1,
"bos_token_id": 2,
"rope_parameters": {
"full_attention": {
"rope_theta": 1000000.0,
"rope_type": "default"
},
"sliding_attention": {
"rope_theta": 10000.0,
"rope_type": "default"
}
},
"attention_bias": false,
"attention_dropout": 0.0,
"sliding_window": 512,
"layer_types": [
"sliding_attention",
"full_attention"
],
"hidden_size_per_layer_input": 512,
"embedding_dim": 24,
"_name_or_path": "",
"model_type": "embedding_gemma2_text",
"output_attentions": false,
"per_layer_config": {
"1": {
"head_dim": 512,
"num_key_value_heads": 1
}
}
},
"vision_config": {
"architectures": null,
"output_hidden_states": false,
"return_dict": true,
"dtype": "bfloat16",
"chunk_size_feed_forward": 0,
"is_encoder_decoder": false,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1"
},
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1
},
"problem_type": null,
"hidden_size": 64,
"intermediate_size": 128,
"num_hidden_layers": 2,
"num_attention_heads": 4,
"num_key_value_heads": 4,
"head_dim": 16,
"hidden_activation": "gelu_pytorch_tanh",
"rms_norm_eps": 1e-06,
"max_position_embeddings": 131072,
"attention_bias": false,
"attention_dropout": 0.0,
"rope_parameters": {
"rope_theta": 100.0,
"rope_type": "axial"
},
"pooling_kernel_size": 3,
"patch_size": 16,
"position_embedding_size": 10240,
"use_clipped_linears": false,
"standardize": false,
"initializer_range": 0.02,
"_name_or_path": "",
"default_output_length": 280,
"global_head_dim": 64,
"model_type": "gemma4_vision",
"output_attentions": false
},
"audio_config": {
"architectures": null,
"output_hidden_states": false,
"return_dict": true,
"dtype": "bfloat16",
"chunk_size_feed_forward": 0,
"is_encoder_decoder": false,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1"
},
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1
},
"problem_type": null,
"hidden_size": 64,
"num_hidden_layers": 2,
"num_attention_heads": 4,
"hidden_act": "silu",
"subsampling_conv_channels": [
128,
32
],
"conv_kernel_size": 5,
"residual_weight": 0.5,
"attention_chunk_size": 12,
"attention_context_left": 13,
"attention_context_right": 0,
"attention_logit_cap": 50.0,
"attention_invalid_logits_value": -1000000000.0,
"use_clipped_linears": true,
"rms_norm_eps": 1e-06,
"gradient_clipping": 10000000000.0,
"output_proj_dims": 1536,
"initializer_range": 0.02,
"_name_or_path": "",
"model_type": "gemma4_audio",
"output_attentions": false
},
"boi_token_id": 255999,
"eoi_token_id": 258882,
"image_token_id": 258880,
"video_token_id": 258884,
"boa_token_id": 256000,
"eoa_token_index": 258883,
"audio_token_id": 258881,
"initializer_range": 0.02,
"model_type": "embedding_gemma2",
"vision_soft_tokens_per_image": 280,
"output_attentions": false,
"dtype": "bfloat16"
},
"covers": {
"google/embeddinggemma-2": "f9972ac22e696a105c482202646979e33e4b5a1e"
},
"built_from": {
"repo": "google/embeddinggemma-2",
"revision": "f9972ac22e696a105c482202646979e33e4b5a1e"
},
"seed": 17,
"dtype": "bfloat16",
"builder": "example/build_fixtures.py",
"builder_revision": "e8034a51dc828ccfb8c6ad3d765dd2275181f0b8",
"recipe": {
"seed": 17,
"model_class": "EmbeddingGemma2Model"
},
"shrink": "rule",
"model_class": "EmbeddingGemma2Model",
"parameters": 19313504,
"missing_parameters": [],
"serialization": {
"file": "model.safetensors",
"reverted_conversions": []
},
"tester_sizes": {
"source": "https://raw.githubusercontent.com/huggingface/transformers/main/tests/models/embedding_gemma2/test_modeling_embedding_gemma2.py",
"fields": {
"text_config.embedding_dim": 24
}
},
"checkpoint_inspection": "parameter headers/indexes; bounded binary fallback",
"numeric_health_scope": "CPU float32/bfloat16/float16; autocast for low-precision losses",
"captured": {
"call": "training step",
"inputs": [
"attention_mask",
"input_ids"
],
"class": "EmbeddingGemma2Model"
},
"weights": "random",
"gate": {
"random": {
"pass": true,
"accepted_insensitive": {},
"threshold": 0.001,
"deltas": {
"input_ids:roll_batch": 0.634765625,
"input_ids:reverse": 0.6630859375,
"attention_mask:all_ones": 0.169921875
},
"weakest": "attention_mask:all_ones",
"no_perturbation_defined": [],
"not_a_forward_parameter": []
}
},
"numeric_health": {
"float32": {
"outputs_finite": true,
"finite": true
},
"bfloat16": {
"outputs_finite": true,
"finite": true
},
"float16": {
"outputs_finite": true,
"finite": true
}
},
"copied_unchanged": [
"chat_template.jinja",
"config_sentence_transformers.json",
"modules.json",
"preprocessor_config.json",
"processor_config.json",
"sentence_bert_config.json",
"tokenizer.json",
"tokenizer.model",
"tokenizer_config.json"
],
"environment": {
"python": "3.12.12",
"machine": "arm64",
"torch": "2.14.1",
"transformers": "5.22.0.dev0",
"tokenizers": "0.23.3"
},
"sha256": {
"chat_template.jinja": "4b852efc0b9960283e735363331e6f325b33bc74bdbaa076f595bc4e9b94d85e",
"config.json": "e5f955a0f142ded3ff723e68d7ad4890b8a846a3793afa0d3f728bfa8451c85a",
"config_sentence_transformers.json": "031e56a498d33c349ab489a21885bcfe25b4fcba841149dc99e1e90d4a7c28f5",
"model.safetensors": "182aae3bfd39973ceb4b21c82d6122f27c35dfe318f4e61f9a43bacffdcc9b6e",
"modules.json": "3d02572a0455b832de67fb8e63a54981bc7e8b46e337c95e917bd8122a533bfd",
"preprocessor_config.json": "ea2ae257e901064abdd98dceb19f2b0da06af600bed15e0f99f5c85c37ee9d78",
"processor_config.json": "168f6a08522f3ce5dea596d94d003af2fd691742d4f41fe1f9d8cce76bfbf69c",
"sentence_bert_config.json": "b1bcd9f2dce3ae863b359e87d0710b5dbc3314a59ecb4e2f97c7778fc8e4b228",
"tokenizer.json": "4d777ef5bdc1aa36227abdfb77c3e49e7b9c892d16e1b6bda41c393504828be4",
"tokenizer.model": "e594c8a90eb08d8bda498ff4747977dc827ae0c3c56b5c0d41a605a22d02ef03",
"tokenizer_config.json": "17bd5d6e9364ca49a534e1502076593317c298d4a663623091ed45388f004874"
}
}