Update base.py

run-llama · Feb 6, 2024 · d08422a · d08422a
1 parent 6fa3171
commit d08422a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llama_hub/llama_packs/rag_cli_local/base.py b/llama_hub/llama_packs/rag_cli_local/base.py
@@ -41,7 +41,7 @@ def init_local_rag_cli(
     print("> Chroma collection initialized")
     llm = Ollama(model=llm_model_name, request_timeout=30.0)
     print("> LLM initialized")
-    embed_model = HuggingFaceEmbedding(model_name=embed_model_name)
+    embed_model = HuggingFaceEmbedding(model_name=embed_model_name, pooling="mean")
     print("> Embedding model initialized")
 
     ingestion_pipeline = IngestionPipeline(