samwjya commited on
Commit ·
646256c
1
Parent(s): fc42a68
change to diff rerannker model
Browse files- reranker.py +3 -3
reranker.py
CHANGED
|
@@ -2,12 +2,12 @@ from concurrent.futures import ThreadPoolExecutor
|
|
| 2 |
from sentence_transformers import CrossEncoder
|
| 3 |
|
| 4 |
|
| 5 |
-
DEFAULT_MODEL = "
|
| 6 |
|
| 7 |
ENSEMBLE_MODELS = [
|
|
|
|
|
|
|
| 8 |
"BAAI/bge-reranker-v2-m3",
|
| 9 |
-
"cross-encoder/ms-marco-MiniLM-L-6-v2",
|
| 10 |
-
"cross-encoder/ms-marco-electra-base",
|
| 11 |
]
|
| 12 |
|
| 13 |
|
|
|
|
| 2 |
from sentence_transformers import CrossEncoder
|
| 3 |
|
| 4 |
|
| 5 |
+
DEFAULT_MODEL = "jinaai/jina-reranker-v2-base-multilingual"
|
| 6 |
|
| 7 |
ENSEMBLE_MODELS = [
|
| 8 |
+
"jinaai/jina-reranker-v2-base-multilingual",
|
| 9 |
+
"Alibaba-NLP/gte-multilingual-reranker-base",
|
| 10 |
"BAAI/bge-reranker-v2-m3",
|
|
|
|
|
|
|
| 11 |
]
|
| 12 |
|
| 13 |
|