metarank/ce-msmarco-MiniLM-L6-v2
By metarank · ↓ 2 · ♥ 0 · Updated 2023-12-01
What's in this repo
This page lists the files in metarank/ce-msmarco-MiniLM-L6-v2, including model.onnx, pytorch_model.onnx.
- 2 ONNX models (.onnx files)
- 10 files, including 2 .json, 2 .onnx, 2 .txt, 1 .md, 1 .py
- Last updated 2023-12-01
ONNX files
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 cross-encoder/mcmarco-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-04-11
Updated: 2023-12-01
View on Hugging Face