Merge pull request #48 from weaviate/add-4-new-models

Add support for 2 new models
weaviate · Feb 1, 2023 · a42a6da · a42a6da
2 parents 7035c9a + 16c4052
commit a42a6da
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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