Nvidia embedding

worker.py

@@ -47,8 +47,9 @@ def init_llm():
     # llm_hub.invoke('foo bar')
 
     #Initialize embeddings using a pre-trained model to represent the text data.
-    # embedddings_model = "sentence-transformers/multi-qa-distilbert-cos-v1"
-    embedddings_model = "sentence-transformers/all-MiniLM-L6-v2"
+    embedddings_model = "sentence-transformers/multi-qa-distilbert-cos-v1"
+    # embedddings_model = "sentence-transformers/all-MiniLM-L6-v2"
+    embedddings_model = "nvidia/NV-Embed-v2"
     embeddings = HuggingFaceInstructEmbeddings(
         model_name=embedddings_model,
         model_kwargs={"device": DEVICE}