From ccd453631ab9276f2744a9471de4cd71bcf387cd Mon Sep 17 00:00:00 2001 From: richinfo-ai Date: Tue, 25 Mar 2025 10:53:59 +0800 Subject: [PATCH 1/2] add richinfoai models add richinfoai models --- mteb/models/overview.py | 2 ++ mteb/models/richinfoai_models.py | 31 +++++++++++++++++++++++++++++++ 2 files changed, 33 insertions(+) create mode 100644 mteb/models/richinfoai_models.py diff --git a/mteb/models/overview.py b/mteb/models/overview.py index dcd4005ca8..a715ab2730 100644 --- a/mteb/models/overview.py +++ b/mteb/models/overview.py @@ -64,6 +64,7 @@ repllama_models, rerankers_custom, rerankers_monot5_based, + richinfoai_models, ru_sentence_models, salesforce_models, sentence_transformers_models, @@ -131,6 +132,7 @@ repllama_models, rerankers_custom, rerankers_monot5_based, + richinfoai_models, ru_sentence_models, salesforce_models, sentence_transformers_models, diff --git a/mteb/models/richinfoai_models.py b/mteb/models/richinfoai_models.py new file mode 100644 index 0000000000..1d59119178 --- /dev/null +++ b/mteb/models/richinfoai_models.py @@ -0,0 +1,31 @@ +from __future__ import annotations + +from mteb.model_meta import ModelMeta +from mteb.models.bge_models import bge_full_data, bge_m3_training_data +from mteb.models.stella_models import stella_zh_datasets + +ritrieve_zh_v1 = ModelMeta( + name="richinfoai/ritrieve_zh_v1", + languages=["zho_Hans"], + open_weights=True, + revision="f8d5a707656c55705027678e311f9202c8ced12c", + release_date="2025-03-25", + n_parameters=int(326 * 1e6), + memory_usage_mb=1242, + embed_dim=1792, + license="mit", + max_tokens=512, + reference="https://huggingface.co/richinfoai/ritrieve_zh_v1", + similarity_fn_name="cosine", + framework=["Sentence Transformers", "PyTorch"], + use_instructions=False, + superseded_by=None, + adapted_from=None, + public_training_code=None, + public_training_data=None, + training_datasets={ + **stella_zh_datasets, + **bge_full_data, + **bge_m3_training_data + } +) \ No newline at end of file From 39ec0fb3394bba6bb2572fef01df87fdc7a67f0f Mon Sep 17 00:00:00 2001 From: richinfo-ai Date: Tue, 25 Mar 2025 16:04:49 +0800 Subject: [PATCH 2/2] format codes by linter format codes by linter --- mteb/models/richinfoai_models.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/mteb/models/richinfoai_models.py b/mteb/models/richinfoai_models.py index 1d59119178..c46df9c378 100644 --- a/mteb/models/richinfoai_models.py +++ b/mteb/models/richinfoai_models.py @@ -23,9 +23,5 @@ adapted_from=None, public_training_code=None, public_training_data=None, - training_datasets={ - **stella_zh_datasets, - **bge_full_data, - **bge_m3_training_data - } -) \ No newline at end of file + training_datasets={**stella_zh_datasets, **bge_full_data, **bge_m3_training_data}, +)