Views
No views yet
| Property | Value |
|---|---|
| Original Model | sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2 |
| Format | LiteRT (.tflite) |
| File Size | 448.2 MB |
| Task | Multilingual Sentence Embeddings (50+ languages) |
| Max Sequence Length | 128 |
| Output Dimension | 384 |
| Pooling Mode | Mean Pooling |
| Metric | Value |
|---|---|
| Inference Latency | 21.3 ms |
| Throughput | 47.0/sec |
| Cosine Similarity vs Original | 1.0000 ✅ |
1import numpy as np
2from ai_edge_litert.interpreter import Interpreter
3from transformers import AutoTokenizer
4
5# Load model and tokenizer
6interpreter = Interpreter(model_path="sentence-transformers_paraphrase-multilingual-MiniLM-L12-v2.tflite")
7interpreter.allocate_tensors()
8input_details = interpreter.get_input_details()
9output_details = interpreter.get_output_details()
10
11tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
12
13def get_embedding(text: str) -> np.ndarray:
14 """Get sentence embedding for input text."""
15 encoded = tokenizer(
16 text,
17 padding="max_length",
18 max_length=128,
19 truncation=True,
20 return_tensors="np"
21 )
22
23 interpreter.set_tensor(input_details[0]["index"], encoded["input_ids"].astype(np.int64))
24 interpreter.set_tensor(input_details[1]["index"], encoded["attention_mask"].astype(np.int64))
25 interpreter.invoke()
26
27 return interpreter.get_tensor(output_details[0]["index"])[0]
28
29# Example
30embedding = get_embedding("Hello, world!")
31print(f"Embedding shape: {embedding.shape}") # (384,)sentence-transformers_paraphrase-multilingual-MiniLM-L12-v2.tflite - The LiteRT model file1@inproceedings{reimers-2020-multilingual-sentence-bert,
2 title = "Making Monolingual Sentence Embeddings Multilingual using Knowledge Distillation",
3 author = "Reimers, Nils and Gurevych, Iryna",
4 booktitle = "EMNLP 2020",
5 year = "2020",
6 url = "https://arxiv.org/abs/2004.09813",
7}