From 43c6ddb262e9380916de4d5e119b68aa2eb9a153 Mon Sep 17 00:00:00 2001 From: Marcin Antas Date: Mon, 22 Apr 2024 07:23:42 +0200 Subject: [PATCH] Add support for Snowflake's Arctic models --- .github/workflows/main.yaml | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/.github/workflows/main.yaml b/.github/workflows/main.yaml index 83e39e7..e4c66ac 100644 --- a/.github/workflows/main.yaml +++ b/.github/workflows/main.yaml @@ -103,6 +103,30 @@ jobs: - model_name: mixedbread-ai/mxbai-embed-large-v1 model_tag_name: mixedbread-ai-mxbai-embed-large-v1 onnx_runtime: true + - model_name: Snowflake/snowflake-arctic-embed-xs + model_tag_name: snowflake-snowflake-arctic-embed-xs + onnx_runtime: false + - model_name: Snowflake/snowflake-arctic-embed-xs + model_tag_name: snowflake-snowflake-arctic-embed-xs + onnx_runtime: true + - model_name: Snowflake/snowflake-arctic-embed-s + model_tag_name: snowflake-snowflake-arctic-embed-s + onnx_runtime: false + - model_name: Snowflake/snowflake-arctic-embed-s + model_tag_name: snowflake-snowflake-arctic-embed-s + onnx_runtime: true + - model_name: Snowflake/snowflake-arctic-embed-m + model_tag_name: snowflake-snowflake-arctic-embed-m + onnx_runtime: false + - model_name: Snowflake/snowflake-arctic-embed-m + model_tag_name: snowflake-snowflake-arctic-embed-m + onnx_runtime: true + - model_name: Snowflake/snowflake-arctic-embed-l + model_tag_name: snowflake-snowflake-arctic-embed-l + onnx_runtime: false + - model_name: Snowflake/snowflake-arctic-embed-l + model_tag_name: snowflake-snowflake-arctic-embed-l + onnx_runtime: true env: LOCAL_REPO: transformers-inference REMOTE_REPO: semitechnologies/transformers-inference