Feature Extraction
sentence-transformers
ONNX
English
sentence-similarity
Inference Endpoints
shuttie commited on
Commit
9f30837
1 Parent(s): 6f23931

add readme and model card

Browse files
Files changed (2) hide show
  1. README.md +65 -1
  2. requirements.txt +2 -1
README.md CHANGED
@@ -1,2 +1,66 @@
1
- # ONNX port of sentence-transormers/all-MiniLM-L6-v2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - sentence-transformers
4
+ - feature-extraction
5
+ - sentence-similarity
6
+ language: en
7
+ license: apache-2.0
8
+ datasets:
9
+ - s2orc
10
+ - flax-sentence-embeddings/stackexchange_xml
11
+ - ms_marco
12
+ - gooaq
13
+ - yahoo_answers_topics
14
+ - code_search_net
15
+ - search_qa
16
+ - eli5
17
+ - snli
18
+ - multi_nli
19
+ - wikihow
20
+ - natural_questions
21
+ - trivia_qa
22
+ - embedding-data/sentence-compression
23
+ - embedding-data/flickr30k-captions
24
+ - embedding-data/altlex
25
+ - embedding-data/simple-wiki
26
+ - embedding-data/QQP
27
+ - embedding-data/SPECTER
28
+ - embedding-data/PAQ_pairs
29
+ - embedding-data/WikiAnswers
30
 
31
+ ---
32
+
33
+ # ONNX version of sentence-transormers/all-MiniLM-L6-v2
34
+
35
+ 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](https://github.com/metarank/metarank) re-ranker
36
+ to do semantic similarity.
37
+
38
+ Check out the [main Metarank docs](https://docs.metarank.ai) on how to configure it.
39
+
40
+ TLDR:
41
+ ```yaml
42
+ - type: field_match
43
+ name: title_query_match
44
+ rankingField: ranking.query
45
+ itemField: item.title
46
+ distance: cos
47
+ method:
48
+ type: bert
49
+ model: metarank/all-MiniLM-L6-v2
50
+ ```
51
+
52
+ ## Building the model
53
+
54
+ ```shell
55
+ $> pip install -r requirements.txt
56
+ $> python convert.py
57
+
58
+ ============= Diagnostic Run torch.onnx.export version 2.0.0+cu117 =============
59
+ verbose: False, log level: Level.ERROR
60
+ ======================= 0 NONE 0 NOTE 0 WARNING 0 ERROR ========================
61
+
62
+ ```
63
+
64
+ ## License
65
+
66
+ Apache 2.0
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  sentence-transformers==2.2.2
2
  torch==2.0.0
3
- onnx=1.13.1
 
 
1
  sentence-transformers==2.2.2
2
  torch==2.0.0
3
+ onnx==1.13.1
4
+ huggingface_hub==0.13.3