Feature Extraction
Transformers
Safetensors
sentence-transformers
multilingual
embedding_gemma2
embedding
multimodal-embedding
multimodal
vision
audio
video
image-feature-extraction
audio-feature-extraction
video-feature-extraction
sentence-similarity
Instructions to use google/embeddinggemma-2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use google/embeddinggemma-2 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="google/embeddinggemma-2")# pip install -U transformers accelerate # Load model directly from transformers import AutoProcessor, AutoModel processor = AutoProcessor.from_pretrained("google/embeddinggemma-2") model = AutoModel.from_pretrained("google/embeddinggemma-2", device_map="auto") - sentence-transformers
How to use google/embeddinggemma-2 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("google/embeddinggemma-2") sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - Notebooks
- Google Colab
- Kaggle
Update README.md
#21
by hschechter - opened
README.md
CHANGED
|
@@ -16,7 +16,7 @@ tags:
|
|
| 16 |
- sentence-similarity
|
| 17 |
language:
|
| 18 |
- multilingual
|
| 19 |
-
library_name: transformers
|
| 20 |
---
|
| 21 |
|
| 22 |
<div align="center">
|
|
@@ -79,12 +79,12 @@ EmbeddingGemma 2 was evaluated across text, code, vision, visual document, video
|
|
| 79 |
| :---- | :---- | :---- | :---- | :---- |
|
| 80 |
| *Text* | Massive Text Embedding Benchmark (MTEB, multilingual, v2) | Mean(Task), Multiple | 61.36 | 61.15 |
|
| 81 |
| | Massive Text Embedding Benchmark (MTEB, code, v1) | Mean(Task), NDCG@10 | 78.68 | 68.76 |
|
| 82 |
-
| *Image* | Massive Image Embedding Benchmark (MIEB, lite) | Mean(TaskType), Multiple |
|
| 83 |
| | Massive Multimodal Embedding Benchmark (MMEB v2 \- Image) | Mean(Task), Hit@1 | 57.28 | \- |
|
| 84 |
| | Massive Multimodal Embedding Benchmark (MMEB v2 \- VisDoc) | Mean(Task), NDCG@5 | 67.84 | \- |
|
| 85 |
| *Video* | Massive Multimodal Embedding Benchmark (MMEB v2 \- Video) | Mean(Task), Hit@1 | 50.67 | \- |
|
| 86 |
| *Audio* | Massive Sound Embedding Benchmark (MSEB, Retrieval) | Mean(Task), MRR@10 | 69.54 | \- |
|
| 87 |
-
| | Massive Audio Embedding Benchmark (MAEB) *Hugging Face* | Mean(Task), Multiple | 49.
|
| 88 |
|
| 89 |
### Evaluation Results with Vector Truncation
|
| 90 |
|
|
@@ -93,10 +93,10 @@ With this, model users can reduce storage requirements, with minimal quality imp
|
|
| 93 |
|
| 94 |
| Output Dimension | Compression Ratio | MTEB *(multilingual, v2) Mean(Task)* | MTEB *(eng, v2) Mean(Task)* | MTEB *(code, v1) Mean(Task)* | MIEB *(lite) Mean(TaskType)* | MMEB (v2) *Overall* | MSEB *(Retrieval) Mean(Task)* | MAEB *Mean(Task)* |
|
| 95 |
| :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
|
| 96 |
-
| 768d (Full) | 1:1 | 61.36 | 68.46 | 78.68 |
|
| 97 |
-
| 512d | 1:1.5 | 61.17 | 68.41 | 77.24 |
|
| 98 |
-
| 256d | 1:3 | 60.41 | 67.78 | 76.18 |
|
| 99 |
-
| 128d | 1:6 | 57.89 | 65.68 | 71.41 | 59.
|
| 100 |
|
| 101 |
## **Quick Start**
|
| 102 |
|
|
|
|
| 16 |
- sentence-similarity
|
| 17 |
language:
|
| 18 |
- multilingual
|
| 19 |
+
library_name: sentence-transformers
|
| 20 |
---
|
| 21 |
|
| 22 |
<div align="center">
|
|
|
|
| 79 |
| :---- | :---- | :---- | :---- | :---- |
|
| 80 |
| *Text* | Massive Text Embedding Benchmark (MTEB, multilingual, v2) | Mean(Task), Multiple | 61.36 | 61.15 |
|
| 81 |
| | Massive Text Embedding Benchmark (MTEB, code, v1) | Mean(Task), NDCG@10 | 78.68 | 68.76 |
|
| 82 |
+
| *Image* | Massive Image Embedding Benchmark (MIEB, lite) | Mean(TaskType), Multiple | 65.67 | \- |
|
| 83 |
| | Massive Multimodal Embedding Benchmark (MMEB v2 \- Image) | Mean(Task), Hit@1 | 57.28 | \- |
|
| 84 |
| | Massive Multimodal Embedding Benchmark (MMEB v2 \- VisDoc) | Mean(Task), NDCG@5 | 67.84 | \- |
|
| 85 |
| *Video* | Massive Multimodal Embedding Benchmark (MMEB v2 \- Video) | Mean(Task), Hit@1 | 50.67 | \- |
|
| 86 |
| *Audio* | Massive Sound Embedding Benchmark (MSEB, Retrieval) | Mean(Task), MRR@10 | 69.54 | \- |
|
| 87 |
+
| | Massive Audio Embedding Benchmark (MAEB) *Hugging Face* | Mean(Task), Multiple | 49.87 | \- |
|
| 88 |
|
| 89 |
### Evaluation Results with Vector Truncation
|
| 90 |
|
|
|
|
| 93 |
|
| 94 |
| Output Dimension | Compression Ratio | MTEB *(multilingual, v2) Mean(Task)* | MTEB *(eng, v2) Mean(Task)* | MTEB *(code, v1) Mean(Task)* | MIEB *(lite) Mean(TaskType)* | MMEB (v2) *Overall* | MSEB *(Retrieval) Mean(Task)* | MAEB *Mean(Task)* |
|
| 95 |
| :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
|
| 96 |
+
| 768d (Full) | 1:1 | 61.36 | 68.46 | 78.68 | 65.67 | 59.01 | 69.54 | 49.87 |
|
| 97 |
+
| 512d | 1:1.5 | 61.17 | 68.41 | 77.24 | 65.34 | 58.38 | 69.18 | 49.76 |
|
| 98 |
+
| 256d | 1:3 | 60.41 | 67.78 | 76.18 | 64.16 | 56.24 | 66.76 | 49.13 |
|
| 99 |
+
| 128d | 1:6 | 57.89 | 65.68 | 71.41 | 59.97 | 45.65 | 56.71 | 47.28 |
|
| 100 |
|
| 101 |
## **Quick Start**
|
| 102 |
|