Merge pull request #75 from weaviate/add-support-for-baai-bge-m3-onnx…

…-image Add support for BAAI/bge-m3 model with ONNX
weaviate · Mar 5, 2024 · 061c110 · 061c110
2 parents 55f01cd + 9c2d2ca
commit 061c110
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 15 deletions.
diff --git a/.github/workflows/main.yaml b/.github/workflows/main.yaml
@@ -90,6 +90,9 @@ jobs:
           - model_name: BAAI/bge-base-en-v1.5
             model_tag_name: baai-bge-base-en-v1.5
             onnx_runtime: true
+          - model_name: BAAI/bge-m3
+            model_tag_name: baai-bge-m3
+            onnx_runtime: true
           - model_name: sentence-transformers/all-MiniLM-L6-v2
             model_tag_name: sentence-transformers-all-MiniLM-L6-v2
             onnx_runtime: true

diff --git a/requirements-test.txt b/requirements-test.txt
@@ -1,12 +1,12 @@
-requests==2.28.2
-transformers==4.27.2
-fastapi==0.95.0
-uvicorn==0.21.1
+requests==2.31.0
+transformers==4.38.2
+fastapi==0.110.0
+uvicorn==0.27.1
 nltk==3.8.1
 torch==2.0.1
-sentencepiece==0.1.97
+sentencepiece==0.2.0
 sentence-transformers==2.2.2
-optimum==1.13.2
-onnxruntime==1.16.1
-onnx==1.14.1
+optimum==1.17.1
+onnxruntime==1.17.1
+onnx==1.15.0
 pytest
diff --git a/requirements.txt b/requirements.txt
@@ -1,10 +1,10 @@
-transformers==4.34.1
-fastapi==0.103.2
-uvicorn==0.23.2
+transformers==4.38.2
+fastapi==0.110.0
+uvicorn==0.27.1
 nltk==3.8.1
 torch==2.0.1
-sentencepiece==0.1.99
+sentencepiece==0.2.0
 sentence-transformers==2.2.2
-optimum==1.13.2
-onnxruntime==1.16.1
-onnx==1.14.1
+optimum==1.17.1
+onnxruntime==1.17.1
+onnx==1.15.0