From 23af82e30db856393c1f456ff7c60cc0b9dc944e Mon Sep 17 00:00:00 2001
From: Yasmine MHIRI <42073781+YasmineMh@users.noreply.github.com>
Date: Sat, 3 Feb 2024 00:34:55 +0100
Subject: [PATCH] fix: model_name validation error and nomic embed model name
(#10421)
---
llama_index/embeddings/nomic.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/llama_index/embeddings/nomic.py b/llama_index/embeddings/nomic.py
index f76876b05..555e66753 100644
--- a/llama_index/embeddings/nomic.py
+++ b/llama_index/embeddings/nomic.py
@@ -32,7 +32,7 @@ class NomicEmbedding(BaseEmbedding):
def __init__(
self,
- model_name: str = "nomic-text-embed-v1",
+ model_name: str = "nomic-embed-text-v1",
embed_batch_size: int = 32,
api_key: Optional[str] = None,
callback_manager: Optional[CallbackManager] = None,
@@ -57,6 +57,7 @@ class NomicEmbedding(BaseEmbedding):
if api_key is not None:
nomic.cli.login(api_key)
super().__init__(
+ model_name=model_name,
embed_batch_size=embed_batch_size,
callback_manager=callback_manager,
_model=embed,
--
GitLab