binhcode25 commited on
Commit
e305ac0
·
verified ·
1 Parent(s): 98654fa

Add new SentenceTransformer model.

Browse files
README.md ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: light-embed
3
+ pipeline_tag: sentence-similarity
4
+ tags:
5
+ - sentence-transformers
6
+ - feature-extraction
7
+ - sentence-similarity
8
+
9
+ ---
10
+
11
+ # sbert-paraphrase-MiniLM-L6-v2-onnx
12
+
13
+ This is the ONNX version of the Sentence Transformers model sentence-transformers/paraphrase-MiniLM-L6-v2 for sentence embedding, optimized for speed and lightweight performance. By utilizing onnxruntime and tokenizers instead of heavier libraries like sentence-transformers and transformers, this version ensures a smaller library size and faster execution. Below are the details of the model:
14
+ - Base model: sentence-transformers/paraphrase-MiniLM-L6-v2
15
+ - Embedding dimension: 384
16
+ - Max sequence length: 128
17
+ - File size on disk: 0.08 GB
18
+
19
+ This ONNX model consists all components in the original sentence transformer model:
20
+ Transformer, Pooling
21
+
22
+ <!--- Describe your model here -->
23
+
24
+ ## Usage (LightEmbed)
25
+
26
+ Using this model becomes easy when you have [LightEmbed](https://www.light-embed.net) installed:
27
+
28
+ ```
29
+ pip install -U light-embed
30
+ ```
31
+
32
+ Then you can use the model like this:
33
+
34
+ ```python
35
+ from light_embed import TextEmbedding
36
+ sentences = ["This is an example sentence", "Each sentence is converted"]
37
+
38
+ model = TextEmbedding('sentence-transformers/paraphrase-MiniLM-L6-v2')
39
+ embeddings = model.encode(sentences)
40
+ print(embeddings)
41
+ ```
42
+
43
+ ## Citing & Authors
44
+
45
+ Binh Nguyen / binhcode25@gmail.com
config.json CHANGED
@@ -1,10 +1,9 @@
1
  {
2
- "_name_or_path": "/Users/Shared/Libs/sentence_transformers/models--sentence-transformers--paraphrase-MiniLM-L6-v2/snapshots/3bf4ae7445aa77c8daaef06518dd78baffff53c9",
3
  "architectures": [
4
  "BertModel"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
- "classifier_dropout": null,
8
  "gradient_checkpointing": false,
9
  "hidden_act": "gelu",
10
  "hidden_dropout_prob": 0.1,
@@ -18,9 +17,8 @@
18
  "num_hidden_layers": 6,
19
  "pad_token_id": 0,
20
  "position_embedding_type": "absolute",
21
- "torch_dtype": "float32",
22
- "transformers_version": "4.37.2",
23
  "type_vocab_size": 2,
24
  "use_cache": true,
25
  "vocab_size": 30522
26
- }
 
1
  {
2
+ "_name_or_path": "sentence-transformers/paraphrase-MiniLM-L6-v2",
3
  "architectures": [
4
  "BertModel"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
 
7
  "gradient_checkpointing": false,
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.1,
 
17
  "num_hidden_layers": 6,
18
  "pad_token_id": 0,
19
  "position_embedding_type": "absolute",
20
+ "transformers_version": "4.7.0",
 
21
  "type_vocab_size": 2,
22
  "use_cache": true,
23
  "vocab_size": 30522
24
+ }
config_sentence_transformers.json CHANGED
@@ -3,7 +3,5 @@
3
  "sentence_transformers": "2.0.0",
4
  "transformers": "4.7.0",
5
  "pytorch": "1.9.0+cu102"
6
- },
7
- "prompts": {},
8
- "default_prompt_name": null
9
  }
 
3
  "sentence_transformers": "2.0.0",
4
  "transformers": "4.7.0",
5
  "pytorch": "1.9.0+cu102"
6
+ }
 
 
7
  }
model.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9edf23f5ee749d16f3e59c9220bf8ae3e132e5c8850b880f95c0f96069554d84
3
- size 90979455
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7073175d5f2eb39ae4c6e0363753be2b20b1d427419a6fdc67c2a944f28aed0c
3
+ size 90445337
model_description.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "base_model": "sentence-transformers/paraphrase-MiniLM-L6-v2",
3
+ "embedding_dim": 384,
4
+ "max_seq_length": 128,
5
+ "model_file_size (GB)": 0.08
6
+ }
modules.json CHANGED
@@ -2,13 +2,11 @@
2
  {
3
  "idx": 0,
4
  "name": "0",
5
- "path": "",
6
  "type": "sentence_transformers.models.Transformer"
7
  },
8
  {
9
  "idx": 1,
10
  "name": "1",
11
- "path": "1_Pooling",
12
  "type": "sentence_transformers.models.Pooling"
13
  }
14
  ]
 
2
  {
3
  "idx": 0,
4
  "name": "0",
 
5
  "type": "sentence_transformers.models.Transformer"
6
  },
7
  {
8
  "idx": 1,
9
  "name": "1",
 
10
  "type": "sentence_transformers.models.Pooling"
11
  }
12
  ]