From f4261592066454f1b9b44f6bce59b0c1ea2e9ac7 Mon Sep 17 00:00:00 2001
From: Kenneth Enevoldsen <kennethcenevoldsen@gmail.com>
Date: Wed, 1 Jan 2025 14:31:54 +0100
Subject: [PATCH] fix: Pass trust_remote_code=True to CPM model (#1669)

Fixes #1651
---
 mteb/models/misc_models.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/mteb/models/misc_models.py b/mteb/models/misc_models.py
index 61dc549b15..49cb0105d1 100644
--- a/mteb/models/misc_models.py
+++ b/mteb/models/misc_models.py
@@ -1,6 +1,8 @@
 from __future__ import annotations
 
-from mteb.model_meta import ModelMeta
+from functools import partial
+
+from mteb.model_meta import ModelMeta, sentence_transformers_loader
 
 Haon_Chen__speed_embedding_7b_instruct = ModelMeta(
     name="Haon-Chen/speed-embedding-7b-instruct",
@@ -1535,11 +1537,16 @@
     superseded_by=None,
 )
 openbmb__MiniCPM_Embedding = ModelMeta(
+    loader=partial(  # type: ignore
+        sentence_transformers_loader,
+        name="openbmb/MiniCPM-Embedding",
+        revision="c0cb2de33fb366e17c30f9d53142ff11bc18e049",
+        trust_remote_code=True,
+    ),
     name="openbmb/MiniCPM-Embedding",
     revision="c0cb2de33fb366e17c30f9d53142ff11bc18e049",
     release_date="2024-09-04",
     languages=["zho_Hans", "eng_Latn"],
-    loader=None,
     n_parameters=2724880896,
     memory_usage=None,
     max_tokens=512.0,