Bombek1/all-MiniLM-L6-v2-litert

all-MiniLM-L6-v2 - LiteRT

This is a LiteRT (formerly TensorFlow Lite) conversion of sentence-transformers/all-MiniLM-L6-v2 for efficient on-device inference.

Model Details

Property	Value
Original Model	sentence-transformers/all-MiniLM-L6-v2
Format	LiteRT (.tflite)
File Size	85.8 MB
Task	Sentence Embeddings / Semantic Similarity
Max Sequence Length	128
Output Dimension	384
Pooling Mode	Mean Pooling

Performance

Benchmarked on AMD CPU (WSL2):

Metric	Value
Inference Latency	10.6 ms
Throughput	94.7/sec
Cosine Similarity vs Original	1.0000 ✅

Quick Start

import numpy as np
from ai_edge_litert.interpreter import Interpreter
from transformers import AutoTokenizer

# Load model and tokenizer
interpreter = Interpreter(model_path="sentence-transformers_all-MiniLM-L6-v2.tflite")
interpreter.allocate_tensors()
input_details = interpreter.get_input_details()
output_details = interpreter.get_output_details()

tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/all-MiniLM-L6-v2")

def get_embedding(text: str) -> np.ndarray:
    """Get sentence embedding for input text."""
    encoded = tokenizer(
        text,
        padding="max_length",
        max_length=128,
        truncation=True,
        return_tensors="np"
    )

    interpreter.set_tensor(input_details[0]["index"], encoded["input_ids"].astype(np.int64))
    interpreter.set_tensor(input_details[1]["index"], encoded["attention_mask"].astype(np.int64))
    interpreter.invoke()

    return interpreter.get_tensor(output_details[0]["index"])[0]

# Example
embedding = get_embedding("Hello, world!")
print(f"Embedding shape: {embedding.shape}")  # (384,)

Files

• sentence-transformers_all-MiniLM-L6-v2.tflite - The LiteRT model file

Conversion Details

• Conversion Tool: ai-edge-torch
• Conversion Date: 2026-01-12
• Source Framework: PyTorch → LiteRT
• Validation: Cosine similarity 1.0000 vs original

Intended Use

• Mobile Applications: On-device semantic search, RAG systems
• Edge Devices: IoT, embedded systems, Raspberry Pi
• Offline Processing: Privacy-preserving inference
• Low-latency Applications: Real-time processing

Limitations

• Fixed sequence length (128 tokens)
• CPU inference (GPU delegate requires setup)
• Tokenizer loaded separately from original model
• Float32 precision

License

This model inherits the license from the original:

• License: Apache 2.0 (source)

Citation

@inproceedings{reimers-2019-sentence-bert,
    title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
    author = "Reimers, Nils and Gurevych, Iryna",
    booktitle = "EMNLP 2019",
    year = "2019",
    url = "https://arxiv.org/abs/1908.10084",
}

Acknowledgments

• Original model by sentence-transformers
• Conversion using ai-edge-torch

---

Converted by Bombek1