diff --git a/.github/workflows/main.yaml b/.github/workflows/main.yaml index 3b43c2d..d371133 100644 --- a/.github/workflows/main.yaml +++ b/.github/workflows/main.yaml @@ -78,14 +78,14 @@ jobs: - model_name: biu-nlp/abstract-sim-query model_tag_name: biu-nlp-abstract-sim-query onnx_runtime: false - - model_name: BAAI/bge-small-en - model_tag_name: baai-bge-small-en - onnx_runtime: true - model_name: BAAI/bge-small-en-v1.5 model_tag_name: baai-bge-small-en-v1.5 - onnx_runtime: true - - model_name: BAAI/bge-base-en - model_tag_name: baai-bge-base-en + onnx_runtime: false + - model_name: BAAI/bge-base-en-v1.5 + model_tag_name: baai-bge-base-en-v1.5 + onnx_runtime: false + - model_name: BAAI/bge-small-en-v1.5 + model_tag_name: baai-bge-small-en-v1.5 onnx_runtime: true - model_name: BAAI/bge-base-en-v1.5 model_tag_name: baai-bge-base-en-v1.5 diff --git a/README.md b/README.md index a3b24ad..75f1cad 100644 --- a/README.md +++ b/README.md @@ -39,6 +39,8 @@ The pre-built models include: |`sentence-transformers/gtr-t5-large` ([Info](https://huggingface.co/sentence-transformers/gtr-t5-large))|`semitechnologies/transformers-inference:sentence-transformers-gtr-t5-large`| |`google/flan-t5-base` ([Info](https://huggingface.co/google/flan-t5-base))|`semitechnologies/transformers-inference:sentence-transformers-gtr-t5-base`| |`google/flan-t5-large` ([Info](https://huggingface.co/google/flan-t5-large))|`semitechnologies/transformers-inference:sentence-transformers-gtr-t5-large`| +|`BAAI/bge-small-en-v1.5` ([Info](https://huggingface.co/BAAI/bge-small-en-v1.5))|`semitechnologies/transformers-inference:baai-bge-small-en-v1.5`| +|`BAAI/bge-base-en-v1.5` ([Info](https://huggingface.co/BAAI/bge-base-en-v1.5))|`semitechnologies/transformers-inference:baai-bge-base-en-v1.5`| |DPR Models| |`facebook/dpr-ctx_encoder-single-nq-base` ([Info](https://huggingface.co/facebook/dpr-ctx_encoder-single-nq-base))|`semitechnologies/transformers-inference:facebook-dpr-ctx_encoder-single-nq-base`| |`facebook/dpr-question_encoder-single-nq-base` ([Info](https://huggingface.co/facebook/dpr-question_encoder-single-nq-base))|`semitechnologies/transformers-inference:facebook-dpr-question_encoder-single-nq-base`| @@ -48,9 +50,7 @@ The pre-built models include: |`biu-nlp/abstract-sim-sentence` ([Info](https://huggingface.co/biu-nlp/abstract-sim-sentence))|`semitechnologies/transformers-inference:biu-nlp-abstract-sim-sentence`| |`biu-nlp/abstract-sim-query` ([Info](https://huggingface.co/biu-nlp/abstract-sim-query))|`semitechnologies/transformers-inference:biu-nlp-abstract-sim-query`| |ONNX Models| -|`BAAI/bge-small-en` ([Info](https://huggingface.co/BAAI/bge-small-en))|`semitechnologies/transformers-inference:baai-bge-small-en-onnx`| |`BAAI/bge-small-en-v1.5` ([Info](https://huggingface.co/BAAI/bge-small-en-v1.5))|`semitechnologies/transformers-inference:baai-bge-small-en-v1.5-onnx`| -|`BAAI/bge-base-en` ([Info](https://huggingface.co/BAAI/bge-base-en))|`semitechnologies/transformers-inference:baai-bge-base-en-onnx`| |`BAAI/bge-base-en-v1.5` ([Info](https://huggingface.co/BAAI/bge-base-en-v1.5))|`semitechnologies/transformers-inference:baai-bge-base-en-v1.5-onnx`| |`sentence-transformers/all-MiniLM-L6-v2` ([Info](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2))|`semitechnologies/transformers-inference:sentence-transformers-all-MiniLM-L6-v2-onnx`|