metadata
library_name: onnx
tags:
- sentence-transformers
- sentence-similarity
- feature-extraction
- onnx
- inference4j
license: apache-2.0
pipeline_tag: sentence-similarity
all-MiniLM-L6-v2 — ONNX
ONNX export of all-MiniLM-L6-v2, a sentence-transformers model that maps sentences to a 384-dimensional dense vector space. Fine-tuned on a 1B+ sentence pairs dataset using a self-supervised contrastive learning objective.
Mirrored for use with inference4j, an inference-only AI library for Java.
Original Source
- Repository: sentence-transformers
- License: apache-2.0
Usage with inference4j
try (SentenceTransformer model = SentenceTransformer.fromPretrained("models/all-MiniLM-L6-v2")) {
float[] embedding = model.encode("Hello, world!");
System.out.println("Dimension: " + embedding.length); // 384
}
Model Details
| Property | Value |
|---|---|
| Architecture | MiniLM-L6 (6 layers, 384 hidden) |
| Task | Sentence embeddings / semantic similarity |
| Output dimension | 384 |
| Max sequence length | 256 |
| Training data | 1B+ sentence pairs |
| Original framework | PyTorch (sentence-transformers) |
License
This model is licensed under the Apache License 2.0. Original model by sentence-transformers.