From 9c2d2ca3014dd1b665853e63cb08e17cfe81206a Mon Sep 17 00:00:00 2001 From: Marcin Antas Date: Mon, 4 Mar 2024 21:58:50 +0100 Subject: [PATCH] Add support for BAAI/bge-m3 model with ONNX --- .github/workflows/main.yaml | 3 +++ requirements-test.txt | 16 ++++++++-------- requirements.txt | 14 +++++++------- 3 files changed, 18 insertions(+), 15 deletions(-) diff --git a/.github/workflows/main.yaml b/.github/workflows/main.yaml index d371133..0419302 100644 --- 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 index 87b827e..fa5867a 100644 --- 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 index 012885e..835f458 100644 --- 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