Skip to content

Commit

Permalink
fix: add nomic models
Browse files Browse the repository at this point in the history
  • Loading branch information
KennethEnevoldsen committed Dec 3, 2024
1 parent 5fa7b7b commit b8328da
Showing 1 changed file with 66 additions and 0 deletions.
66 changes: 66 additions & 0 deletions mteb/models/nomic_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand All @@ -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,
)

0 comments on commit b8328da

Please sign in to comment.