Embedding model to distilbert-base-uncased (#201)

Josh-XT · May 3, 2023 · 718e5ef · 718e5ef
1 parent 9920945
commit 718e5ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/AgentLLM.py b/AgentLLM.py
@@ -38,7 +38,7 @@ def __init__(self, agent_name: str = "AgentLLM", primary_objective=None):
         else:
             self.embedding_function = (
                 embedding_functions.SentenceTransformerEmbeddingFunction(
-                    model_name="all-MiniLM-L6-v2"
+                    model_name="distilbert-base-uncased"
                 )
             )
         self.chroma_persist_dir = f"agents/{agent_name}/memories"