From b8328da03ea7c1c047b34490765e079b66d75aa4 Mon Sep 17 00:00:00 2001
From: Kenneth Enevoldsen <kennethcenevoldsen@gmail.com>
Date: Tue, 3 Dec 2024 16:47:09 +0100
Subject: [PATCH] fix: add nomic models

#1515
---
 mteb/models/nomic_models.py | 66 +++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)

diff --git a/mteb/models/nomic_models.py b/mteb/models/nomic_models.py
index 0600f01be..5039afa64 100644
--- a/mteb/models/nomic_models.py
+++ b/mteb/models/nomic_models.py
@@ -87,6 +87,17 @@ def encode(  # type: ignore
     open_weights=True,
     revision="b0753ae76394dd36bcfb912a46018088bca48be0",
     release_date="2024-02-10",  # first commit
+    n_parameters=137_000_000,
+    memory_usage=None,
+    max_tokens=8192,
+    embed_dim=768,
+    license="apache-2.0",
+    reference="https://huggingface.co/nomic-ai/nomic-embed-text-v1.5",
+    similarity_fn_name="cosine",
+    framework=["Sentence Transformers", "PyTorch"],
+    use_instructions=True,
+    adapted_from=None,
+    supersedes="nomic-ai/nomic-embed-text-v1",
 )
 
 nomic_embed_v1 = ModelMeta(
@@ -111,4 +122,59 @@ def encode(  # type: ignore
     similarity_fn_name="cosine",
     framework=["Sentence Transformers", "PyTorch"],
     use_instructions=True,
+    adapted_from=None,
+    supersedes=None,
+)
+
+nomic_embed_v1_ablated = ModelMeta(
+    loader=partial(  # type: ignore
+        NomicWrapper,
+        trust_remote_code=True,
+        model_name="nomic-ai/nomic-embed-text-v1-ablated",
+        revision="7d948905c5d5d3874fa55a925d68e49dbf411e5f",
+        model_prompts=model_prompts,
+    ),
+    name="nomic-ai/nomic-embed-text-v1-ablated",
+    languages=["eng-Latn"],
+    open_weights=True,
+    revision="7d948905c5d5d3874fa55a925d68e49dbf411e5f",
+    release_date="2024-01-15",  # first commit
+    n_parameters=None,
+    memory_usage=None,
+    max_tokens=8192,
+    embed_dim=768,
+    license="apache-2.0",
+    reference="https://huggingface.co/nomic-ai/nomic-embed-text-v1-ablated",
+    similarity_fn_name="cosine",
+    framework=["Sentence Transformers", "PyTorch"],
+    use_instructions=True,
+    adapted_from=None,
+    supersedes=None,
+)
+
+
+nomic_embed_v1_ablated = ModelMeta(
+    loader=partial(  # type: ignore
+        NomicWrapper,
+        trust_remote_code=True,
+        model_name="nomic-ai/nomic-embed-text-v1-unsupervised",
+        revision="b53d557b15ae63852847c222d336c1609eced93c",
+        model_prompts=model_prompts,
+    ),
+    name="nomic-ai/nomic-embed-text-v1-unsupervised",
+    languages=["eng-Latn"],
+    open_weights=True,
+    revision="b53d557b15ae63852847c222d336c1609eced93c",
+    release_date="2024-01-15",  # first commit
+    n_parameters=None,
+    memory_usage=None,
+    max_tokens=8192,
+    embed_dim=768,
+    license="apache-2.0",
+    reference="https://huggingface.co/nomic-ai/nomic-embed-text-v1-unsupervised",
+    similarity_fn_name="cosine",
+    framework=["Sentence Transformers", "PyTorch"],
+    use_instructions=True,
+    adapted_from=None,
+    supersedes=None,
 )