Add new SentenceTransformer model.

README.md

@@ -8,7 +8,7 @@ tags:
 
 ---
 
-# baai-llm-embedder-onnx
+# LightEmbed/baai-llm-embedder-onnx
 
 This is the ONNX version of the Sentence Transformers model BAAI/llm-embedder 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:
 - Base model: BAAI/llm-embedder
@@ -30,17 +30,34 @@ Using this model becomes easy when you have [LightEmbed](https://pypi.org/projec
 pip install -U light-embed
 ```
 
-Then you can use the model like this:
+Then you can use the model using the original model name like this:
 
 ```python
 from light_embed import TextEmbedding
-sentences = ["This is an example sentence", "Each sentence is converted"]
+sentences = [
+	"This is an example sentence",
+	"Each sentence is converted"
+]
 
 model = TextEmbedding('BAAI/llm-embedder')
 embeddings = model.encode(sentences)
 print(embeddings)
 ```
 
+Then you can use the model using onnx model name like this:
+
+```python
+from light_embed import TextEmbedding
+sentences = [
+	"This is an example sentence",
+	"Each sentence is converted"
+]
+
+model = TextEmbedding('LightEmbed/baai-llm-embedder-onnx')
+embeddings = model.encode(sentences)
+print(embeddings)
+```
+
 ## Citing & Authors
 
 Binh Nguyen / [email protected]

model_description.json

@@ -1,5 +1,7 @@
 {
+  "model_name": "LightEmbed/baai-llm-embedder-onnx",
   "base_model": "BAAI/llm-embedder",
+  "model_file": "model.onnx",
   "embedding_dim": 768,
   "max_seq_length": 512,
   "model_file_size (GB)": 0.41