diff --git a/AgentLLM.py b/AgentLLM.py index 88166aa05d92..afd4d8313552 100644 --- 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"