embeddinggemma-2-5bit / validation.json
prince-canuma's picture
Add 5bit MLX conversion of Google EmbeddingGemma 2
261866f verified
Raw History Blame Contribute Delete
2.41 kB
{
"source_model": "google/embeddinggemma-2",
"source_revision": "914f7f89142e33e77833254d9c9b90c3cef7303b",
"mlx_vlm_revision": "3d87e88402f307efbf68e568971aa887ee7d9ed0",
"variant": "5bit",
"hardware": "Apple M3 Ultra",
"mlx_version": "0.32.3",
"transformers_version": "5.18.0.dev0",
"validation_scope": "Six multilingual text inputs and synthetic image, audio, two-frame video, and text+image; numerical smoke checks, not a retrieval benchmark.",
"tensor_count": 1816,
"tensor_dtypes": [
"BF16",
"U32"
],
"weight_bytes": 1132345990,
"cases": {
"audio": {
"shape": [
1,
768
],
"finite": true,
"minimum_cosine_vs_torch_fp32": 0.9924939810290242,
"maximum_absolute_error_vs_torch_fp32": 0.012260988582341231,
"minimum_cosine_vs_source_bf16": 0.9924979822554912,
"minimum_truncated_cosine_vs_torch_fp32": 0.9930662702623838
},
"image": {
"shape": [
1,
768
],
"finite": true,
"minimum_cosine_vs_torch_fp32": 0.9973338720511176,
"maximum_absolute_error_vs_torch_fp32": 0.007859400252991266,
"minimum_cosine_vs_source_bf16": 0.997414237865403,
"minimum_truncated_cosine_vs_torch_fp32": 0.9974431170381493
},
"text": {
"shape": [
6,
768
],
"finite": true,
"minimum_cosine_vs_torch_fp32": 0.9927592110090616,
"maximum_absolute_error_vs_torch_fp32": 0.01339022872619338,
"minimum_cosine_vs_source_bf16": 0.9927331920745692,
"minimum_truncated_cosine_vs_torch_fp32": 0.9932271189043005,
"retrieval_scores_venus_mars": [
0.6715891361236572,
0.8283919095993042
]
},
"text_image": {
"shape": [
1,
768
],
"finite": true,
"minimum_cosine_vs_torch_fp32": 0.9963339983978767,
"maximum_absolute_error_vs_torch_fp32": 0.009012166427111622,
"minimum_cosine_vs_source_bf16": 0.9963695384281007,
"minimum_truncated_cosine_vs_torch_fp32": 0.996486007046607
},
"video": {
"shape": [
1,
768
],
"finite": true,
"minimum_cosine_vs_torch_fp32": 0.9967272726805344,
"maximum_absolute_error_vs_torch_fp32": 0.009255975001225752,
"minimum_cosine_vs_source_bf16": 0.9968479480067542,
"minimum_truncated_cosine_vs_torch_fp32": 0.9967620682846028
}
}
}