Added E5 Mistral

KennethEnevoldsen · Jan 2, 2024 · 01cfb90 · 01cfb90
1 parent 8ef4e39
commit 01cfb90
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/seb/seb_models/e5_models.py b/src/seb/seb_models/e5_models.py
@@ -125,3 +125,19 @@ def create_multilingual_e5_large() -> SebModel:
         loader=partial(E5Wrapper, model_name=hf_name),  # type: ignore
         meta=meta,
     )
+
+
+@models.register("intfloat/e5-mistral-7b-instruct")
+def create_multilingual_e5_mistral_7b_instruct() -> SebModel:
+    hf_name = "intfloat/e5-mistral-7b-instruct"
+    meta = ModelMeta(
+        name=hf_name.split("/")[-1],
+        huggingface_name=hf_name,
+        reference=f"https://huggingface.co/{hf_name}",
+        languages=[],
+        open_source=True,
+    )
+    return SebModel(
+        loader=partial(E5Wrapper, model_name=hf_name),  # type: ignore
+        meta=meta,
+    )