From 5e0c0736d940543392b9da73a81d92b4d14cfca2 Mon Sep 17 00:00:00 2001 From: tutuDoki <871764357@qq.com> Date: Thu, 3 Jul 2025 00:20:19 +0800 Subject: [PATCH 1/4] add listconranker modelmeta --- mteb/models/listconranker.py | 40 ++++++++++++++++++++++++++++++++++++ mteb/models/overview.py | 2 ++ 2 files changed, 42 insertions(+) create mode 100644 mteb/models/listconranker.py diff --git a/mteb/models/listconranker.py b/mteb/models/listconranker.py new file mode 100644 index 0000000000..7b4e184d94 --- /dev/null +++ b/mteb/models/listconranker.py @@ -0,0 +1,40 @@ +from __future__ import annotations + +from mteb.model_meta import ModelMeta + + +listconranker_training_datasets = { + "CMedQAv1-reranking": ["train"], + "CMedQAv2-reranking": ["train"], + "MMarcoReranking": ["train"], + "T2Reranking": ["train"], + # 'Huatuo26M-Lite': ['train'], + # 'MARC': ['train'], + # 'XL-sum-chinese_simplified': ['train'], + # 'CSL': ['train'], +} + +listconranker = ModelMeta( + loader=None, # implemented in v2.0.0 branch + loader_kwargs=dict( + fp_options="float16", + ), + name="ByteDance/ListConRanker", + languages=["zho-Hans"], + open_weights=True, + revision="95ae6a5f422a916bc36520f0f3e198e7d91520a0", + release_date="2024-12-11", + n_parameters=401_000_000, + memory_usage_mb=1242, + similarity_fn_name="cosine", + training_datasets=listconranker_training_datasets, + embed_dim=1024, + license="mit", + max_tokens=512, + reference="https://huggingface.co/ByteDance/ListConRanker", + framework=["PyTorch"], + use_instructions=False, + public_training_code=None, + public_training_data=None, + is_cross_encoder=True, +) \ No newline at end of file diff --git a/mteb/models/overview.py b/mteb/models/overview.py index e7dbf541e2..7e55808304 100644 --- a/mteb/models/overview.py +++ b/mteb/models/overview.py @@ -48,6 +48,7 @@ jina_models, lens_models, linq_models, + listconranker, llm2clip_models, llm2vec_models, misc_models, @@ -122,6 +123,7 @@ jina_clip, lens_models, linq_models, + listconranker, llm2clip_models, llm2vec_models, misc_models, From ec0b7e92be5f4d4d5e6a5894bdc17e9302e7687d Mon Sep 17 00:00:00 2001 From: tutuDoki <871764357@qq.com> Date: Thu, 3 Jul 2025 00:46:59 +0800 Subject: [PATCH 2/4] fix bugs --- mteb/models/listconranker.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/mteb/models/listconranker.py b/mteb/models/listconranker.py index 7b4e184d94..9c00b08b6c 100644 --- a/mteb/models/listconranker.py +++ b/mteb/models/listconranker.py @@ -16,9 +16,6 @@ listconranker = ModelMeta( loader=None, # implemented in v2.0.0 branch - loader_kwargs=dict( - fp_options="float16", - ), name="ByteDance/ListConRanker", languages=["zho-Hans"], open_weights=True, From 79ccd19a782880647466f31d09658f31e74c8dc2 Mon Sep 17 00:00:00 2001 From: tutuDoki <871764357@qq.com> Date: Thu, 3 Jul 2025 01:25:10 +0800 Subject: [PATCH 3/4] use linter --- mteb/models/listconranker.py | 1 - 1 file changed, 1 deletion(-) diff --git a/mteb/models/listconranker.py b/mteb/models/listconranker.py index 9c00b08b6c..1a3102eb99 100644 --- a/mteb/models/listconranker.py +++ b/mteb/models/listconranker.py @@ -2,7 +2,6 @@ from mteb.model_meta import ModelMeta - listconranker_training_datasets = { "CMedQAv1-reranking": ["train"], "CMedQAv2-reranking": ["train"], From 4ac6a6cd5316a7c656ddd74ec10cd80f8986f588 Mon Sep 17 00:00:00 2001 From: Roman Solomatin <36135455+Samoed@users.noreply.github.com> Date: Thu, 3 Jul 2025 16:28:31 +0300 Subject: [PATCH 4/4] lint --- mteb/models/listconranker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mteb/models/listconranker.py b/mteb/models/listconranker.py index 1a3102eb99..3896894b30 100644 --- a/mteb/models/listconranker.py +++ b/mteb/models/listconranker.py @@ -33,4 +33,4 @@ public_training_code=None, public_training_data=None, is_cross_encoder=True, -) \ No newline at end of file +)