File tree Expand file tree Collapse file tree 4 files changed +3
-9
lines changed
tests/models/language/pooling Expand file tree Collapse file tree 4 files changed +3
-9
lines changed Original file line number Diff line number Diff line change 2323# See #19344
2424MTEB_RERANK_TASKS = ["NFCorpus" ]
2525MTEB_RERANK_LANGS = ["en" ]
26- MTEB_RERANK_TOL = 1e -3
26+ MTEB_RERANK_TOL = 2e -3
2727
2828
2929class VllmMtebEncoder (mteb .Encoder ):
Original file line number Diff line number Diff line change 6868 enable_test = False ),
6969 RerankModelInfo ("BAAI/bge-reranker-v2-m3" ,
7070 architecture = "XLMRobertaForSequenceClassification" ,
71- dtype = "float32" ,
7271 enable_test = False )
7372]
7473
Original file line number Diff line number Diff line change 1818]
1919
2020RERANK_MODELS = [
21- RerankModelInfo (
22- "jinaai/jina-reranker-v2-base-multilingual" ,
23- architecture = "XLMRobertaForSequenceClassification" ,
24- dtype = "float32" ,
25- )
21+ RerankModelInfo ("jinaai/jina-reranker-v2-base-multilingual" ,
22+ architecture = "XLMRobertaForSequenceClassification" )
2623]
2724
2825
Original file line number Diff line number Diff line change 1212RERANK_MODELS = [
1313 RerankModelInfo ("Qwen/Qwen3-Reranker-0.6B" ,
1414 architecture = "Qwen3ForSequenceClassification" ,
15- dtype = "float32" ,
1615 enable_test = True ),
1716 RerankModelInfo ("Qwen/Qwen3-Reranker-4B" ,
1817 architecture = "Qwen3ForSequenceClassification" ,
19- dtype = "float32" ,
2018 enable_test = False )
2119]
2220
You can’t perform that action at this time.
0 commit comments