Merge pull request #881 from KoradaCharan/use-minilm-embeddings

Updated Hugging Face embeddings to use all-MiniLM-L6-v2
assafelovic · Oct 4, 2024 · 06a58d8 · 06a58d8
2 parents 4f10f7b + 42f04e1
commit 06a58d8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/gpt_researcher/memory/embeddings.py b/gpt_researcher/memory/embeddings.py
@@ -46,7 +46,10 @@ def __init__(self, embedding_provider, headers=None, **kwargs):
             case "huggingface":
                 from langchain.embeddings import HuggingFaceEmbeddings
 
-                _embeddings = HuggingFaceEmbeddings()
+                # Specifying the Hugging Face embedding model all-MiniLM-L6-v2
+                _embeddings = HuggingFaceEmbeddings(
+                    model_name="sentence-transformers/all-MiniLM-L6-v2"
+                )
 
             case _:
                 raise Exception("Embedding provider not found.")