From 16c4052d3aa7f8101b94fd20d3c562a7dac3731d Mon Sep 17 00:00:00 2001 From: Marcin Antas Date: Wed, 1 Feb 2023 10:42:03 +0100 Subject: [PATCH] Add support for 2 new models Added support for models: - sentence-transformers/all-MiniLM-L6-v2 - sentence-transformers/multi-qa-distilbert-cos-v1 --- .github/workflows/main.yaml | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/.github/workflows/main.yaml b/.github/workflows/main.yaml index 16a2f38..fc69b55 100644 --- a/.github/workflows/main.yaml +++ b/.github/workflows/main.yaml @@ -18,8 +18,8 @@ jobs: strategy: matrix: include: - - model_name: "distilbert-base-uncased" - model_tag_name: "distilbert-base-uncased" + - model_name: distilbert-base-uncased + model_tag_name: distilbert-base-uncased - model_name: sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2 model_tag_name: sentence-transformers-paraphrase-multilingual-MiniLM-L12-v2 - model_name: sentence-transformers/multi-qa-MiniLM-L6-cos-v1 @@ -46,6 +46,10 @@ jobs: model_tag_name: facebook-dpr-ctx_encoder-single-nq-base - model_name: facebook/dpr-question_encoder-single-nq-base model_tag_name: facebook-dpr-question_encoder-single-nq-base + - model_name: sentence-transformers/all-MiniLM-L6-v2 + model_tag_name: sentence-transformers-all-MiniLM-L6-v2 + - model_name: sentence-transformers/multi-qa-distilbert-cos-v1 + model_tag_name: sentence-transformers-multi-qa-distilbert-cos-v1 env: LOCAL_REPO: transformers-inference REMOTE_REPO: semitechnologies/transformers-inference