From 5b79108b5d6df4798b468652fde88083dea79ca8 Mon Sep 17 00:00:00 2001 From: Joshua Lochner Date: Tue, 19 Sep 2023 02:34:55 +0200 Subject: [PATCH] Update list of supported models --- scripts/supported_models.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/scripts/supported_models.py b/scripts/supported_models.py index 928728eaa..c645aee88 100644 --- a/scripts/supported_models.py +++ b/scripts/supported_models.py @@ -77,6 +77,12 @@ 'BAAI/bge-small-en', 'BAAI/bge-base-en', 'BAAI/bge-large-en', + 'BAAI/bge-large-en-v1.5', + 'BAAI/bge-base-en-v1.5', + 'BAAI/bge-small-en-v1.5', + 'BAAI/bge-large-zh-v1.5', + 'BAAI/bge-base-zh-v1.5', + 'BAAI/bge-small-zh-v1.5', 'allenai/scibert_scivocab_uncased', 'SpanBERT/spanbert-large-cased', 'SpanBERT/spanbert-base-cased', @@ -92,9 +98,10 @@ 'nlptown/bert-base-multilingual-uncased-sentiment', 'ProsusAI/finbert', 'unitary/toxic-bert', + 'BAAI/bge-reranker-large', + 'BAAI/bge-reranker-base', ], - # Token classification 'token-classification': [ 'Davlan/bert-base-multilingual-cased-ner-hrl', @@ -133,8 +140,6 @@ ], }, - - 'camembert': { # Feature extraction 'feature-extraction': [