metarank/all-MiniLM-L6-v2
By metarank · ↓ 0 · ♥ 1 · Updated 2023-09-04
What's in this repo
This page lists the files in metarank/all-MiniLM-L6-v2, including pytorch_model.onnx.
- 1 ONNX models (.onnx files)
- 8 files, including 2 .txt, 1 .md, 1 .py, 1 .onnx, 1 .json
- Last updated 2023-09-04
ONNX files
| File | Size | Download |
|---|---|---|
| pytorch_model.onnx | — | Download |
How to use
These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.
README
ONNX version of sentence-transormers/all-MiniLM-L6-v2
This is a sentence-transformers model: It maps sentences & paragraphs to a 384 dimensional dense vector space and can be used for tasks like clustering or semantic search. The ONNX version of this model is made for the Metarank re-ranker to do semantic similarity.
Check out the main Metarank docs on how to configure it.
TLDR:
- type: field_match
name: title_query_match
rankingField: ranking.query
itemField: item.title
distance: cos
method:
type: bert
model: metarank/all-MiniLM-L6-v2
Building the model
$> pip install -r requirements.txt
$> python convert.py
============= Diagnostic Run torch.onnx.export version 2.0.0+cu117 =============
verbose: False, log level: Level.ERROR
======================= 0 NONE 0 NOTE 0 WARNING 0 ERROR ========================
License
Apache 2.0
Model info
Created: 2023-03-30
Updated: 2023-09-04
View on Hugging Face