diff --git a/mteb/models/nvidia_models.py b/mteb/models/nvidia_models.py index 1cfa01cdef..358f8931aa 100644 --- a/mteb/models/nvidia_models.py +++ b/mteb/models/nvidia_models.py @@ -74,7 +74,7 @@ def instruction_template( NV_embed_v2 = ModelMeta( loader=partial( # type: ignore InstructSentenceTransformerWrapper, - model="nvidia/NV-Embed-v2", + model_name="nvidia/NV-Embed-v2", revision="7604d305b621f14095a1aa23d351674c2859553a", instruction_template=instruction_template, trust_remote_code=True, @@ -105,7 +105,7 @@ def instruction_template( NV_embed_v1 = ModelMeta( loader=partial( # type: ignore InstructSentenceTransformerWrapper, - model="nvidia/NV-Embed-v1", + model_name="nvidia/NV-Embed-v1", revision="7604d305b621f14095a1aa23d351674c2859553a", instruction_template=instruction_template, trust_remote_code=True, diff --git a/mteb/models/salesforce_models.py b/mteb/models/salesforce_models.py index d8a99c3559..c9ec0807dd 100644 --- a/mteb/models/salesforce_models.py +++ b/mteb/models/salesforce_models.py @@ -73,7 +73,7 @@ def instruction_template( SFR_Embedding_Code_2B_R = ModelMeta( loader=partial( # type: ignore InstructSentenceTransformerWrapper, - model_name_or_path="Salesforce/SFR-Embedding-Code-2B_R", + model_name="Salesforce/SFR-Embedding-Code-2B_R", instruction_template=instruction_template, attn="cccc", pooling_method="lasttoken",