embeddinggemma-2-6bit / validation.json
prince-canuma's picture
Add 6bit MLX conversion of Google EmbeddingGemma 2
a61f0b6 verified
Raw History Blame Contribute Delete
2.42 kB
{
"source_model": "google/embeddinggemma-2",
"source_revision": "914f7f89142e33e77833254d9c9b90c3cef7303b",
"mlx_vlm_revision": "3d87e88402f307efbf68e568971aa887ee7d9ed0",
"variant": "6bit",
"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": 1166310022,
"cases": {
"audio": {
"shape": [
1,
768
],
"finite": true,
"minimum_cosine_vs_torch_fp32": 0.9982772233943196,
"maximum_absolute_error_vs_torch_fp32": 0.007221069510830777,
"minimum_cosine_vs_source_bf16": 0.9981821675630772,
"minimum_truncated_cosine_vs_torch_fp32": 0.9983461952682761
},
"image": {
"shape": [
1,
768
],
"finite": true,
"minimum_cosine_vs_torch_fp32": 0.9993743690383678,
"maximum_absolute_error_vs_torch_fp32": 0.004241365532904038,
"minimum_cosine_vs_source_bf16": 0.9993957962541856,
"minimum_truncated_cosine_vs_torch_fp32": 0.9994065378612977
},
"text": {
"shape": [
6,
768
],
"finite": true,
"minimum_cosine_vs_torch_fp32": 0.9977411879497886,
"maximum_absolute_error_vs_torch_fp32": 0.007285331596921737,
"minimum_cosine_vs_source_bf16": 0.9978213717051512,
"minimum_truncated_cosine_vs_torch_fp32": 0.9978704412499004,
"retrieval_scores_venus_mars": [
0.677201509475708,
0.8396952152252197
]
},
"text_image": {
"shape": [
1,
768
],
"finite": true,
"minimum_cosine_vs_torch_fp32": 0.9987029577402078,
"maximum_absolute_error_vs_torch_fp32": 0.007021004022041449,
"minimum_cosine_vs_source_bf16": 0.9987498685307801,
"minimum_truncated_cosine_vs_torch_fp32": 0.9987552275220689
},
"video": {
"shape": [
1,
768
],
"finite": true,
"minimum_cosine_vs_torch_fp32": 0.9989877758071022,
"maximum_absolute_error_vs_torch_fp32": 0.0052427304526217675,
"minimum_cosine_vs_source_bf16": 0.999101305340603,
"minimum_truncated_cosine_vs_torch_fp32": 0.9990469201243957
}
}
}