|
|
--- |
|
|
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](https://huggingface.co/sentence-transformers/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](https://github.com/inference4j/inference4j), an inference-only AI library for Java. |
|
|
|
|
|
## Original Source |
|
|
|
|
|
- **Repository:** [sentence-transformers](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) |
|
|
- **License:** apache-2.0 |
|
|
|
|
|
## Usage with inference4j |
|
|
|
|
|
```java |
|
|
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](https://www.apache.org/licenses/LICENSE-2.0). Original model by [sentence-transformers](https://huggingface.co/sentence-transformers). |
|
|
|
