Onnx Support

README.md

@@ -2866,6 +2866,53 @@ sentence_embeddings = torch.nn.functional.normalize(sentence_embeddings, p=2, di
 print("Sentence embeddings:", sentence_embeddings)
 ```
 
+
+#### Usage of the ONNX files
+
+```python
+from optimum.onnxruntime import ORTModelForFeatureExtraction  # type: ignore
+
+import torch
+from transformers import AutoModel, AutoTokenizer
+
+tokenizer = AutoTokenizer.from_pretrained('BAAI/bge-large-en-v1.5')
+model = AutoModel.from_pretrained('BAAI/bge-large-en-v1.5', revision="refs/pr/13")
+model_ort = ORTModelForFeatureExtraction.from_pretrained('BAAI/bge-large-en-v1.5', revision="refs/pr/13",file_name="onnx/model.onnx")
+
+# Sentences we want sentence embeddings for
+sentences = ["样例数据-1", "样例数据-2"]
+
+# Tokenize sentences
+encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')
+# for s2p(short query to long passage) retrieval task, add an instruction to query (not add instruction for passages)
+# encoded_input = tokenizer([instruction + q for q in queries], padding=True, truncation=True, return_tensors='pt')
+
+model_output_ort = model_ort(**encoded_input)
+# Compute token embeddings
+with torch.no_grad():
+    model_output = model(**encoded_input)
+
+# model_output and model_output_ort are identical
+
+```
+
+#### Usage via infinity
+Its also possible to deploy the onnx files with the [infinity_emb](https://github.com/michaelfeil/infinity) pip package.
+```python
+import asyncio
+from infinity_emb import AsyncEmbeddingEngine, EngineArgs
+
+sentences = ["Embed this is sentence via Infinity.", "Paris is in France."]
+engine = AsyncEmbeddingEngine.from_args(
+    EngineArgs(model_name_or_path = "BAAI/bge-large-en-v1.5", device="cpu", engine="optimum" # or engine="torch"
+))
+
+async def main(): 
+    async with engine:
+        embeddings, usage = await engine.embed(sentences=sentences)
+asyncio.run(main())
+```
+
 ### Usage for Reranker
 
 Different from embedding model, reranker uses question and document as input and directly output similarity instead of embedding. 

onnx/model.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9bc579acdba21c253c62a9bf866891355a63ffa3442b52c8a37d75b2ccb91848
+size 435811539