binhcode25 commited on
Commit
b2d7a71
·
verified ·
1 Parent(s): 5e36d95

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-all-MiniLM-L12-v1-onnx
12
+
13
+ This is the ONNX version of the Sentence Transformers model sentence-transformers/all-MiniLM-L12-v1 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/all-MiniLM-L12-v1
15
+ - Embedding dimension: 384
16
+ - Max sequence length: 256
17
+ - File size on disk: 0.12 GB
18
+
19
+ This ONNX model consists all components in the original sentence transformer model:
20
+ Transformer, Pooling, Normalize
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/all-MiniLM-L12-v1')
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,6 @@
1
  {
2
- "_name_or_path": "/Users/Shared/Libs/sentence_transformers/models--sentence-transformers--all-MiniLM-L12-v1/snapshots/0845ca5ddc07524b3196519cf3940ceb86cf106c",
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 +14,8 @@
18
  "num_hidden_layers": 12,
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/all-MiniLM-L12-v1",
 
 
 
3
  "attention_probs_dropout_prob": 0.1,
 
4
  "gradient_checkpointing": false,
5
  "hidden_act": "gelu",
6
  "hidden_dropout_prob": 0.1,
 
14
  "num_hidden_layers": 12,
15
  "pad_token_id": 0,
16
  "position_embedding_type": "absolute",
17
+ "transformers_version": "4.8.2",
 
18
  "type_vocab_size": 2,
19
  "use_cache": true,
20
  "vocab_size": 30522
21
+ }
config_sentence_transformers.json CHANGED
@@ -3,7 +3,5 @@
3
  "sentence_transformers": "2.0.0",
4
  "transformers": "4.6.1",
5
  "pytorch": "1.8.1"
6
- },
7
- "prompts": {},
8
- "default_prompt_name": null
9
  }
 
3
  "sentence_transformers": "2.0.0",
4
  "transformers": "4.6.1",
5
  "pytorch": "1.8.1"
6
+ }
 
 
7
  }
model.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2867954c66b83b4eed959459b6c9f65bc174fe005b77202710c883e20e46ae30
3
- size 133685318
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88bd6374d90bc930e9c2640598cca342e1ba02c3209407f00dedd356ab45a1e0
3
+ size 133203053
model_description.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "base_model": "sentence-transformers/all-MiniLM-L12-v1",
3
+ "embedding_dim": 384,
4
+ "max_seq_length": 256,
5
+ "model_file_size (GB)": 0.12
6
+ }
modules.json CHANGED
@@ -2,19 +2,16 @@
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
  {
15
  "idx": 2,
16
  "name": "2",
17
- "path": "2_Normalize",
18
  "type": "sentence_transformers.models.Normalize"
19
  }
20
  ]
 
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
  {
13
  "idx": 2,
14
  "name": "2",
 
15
  "type": "sentence_transformers.models.Normalize"
16
  }
17
  ]