Update list of supported models

huggingface · Sep 19, 2023 · 5b79108 · 5b79108
1 parent 4245e26
commit 5b79108
Showing 1 changed file with 8 additions and 3 deletions.
diff --git 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': [