From 4bf9cfa51b9e7ebc5afc951ab48e97aee424fd7d Mon Sep 17 00:00:00 2001 From: YangQianli92 <108046369+YangQianli92@users.noreply.github.com> Date: Wed, 17 Apr 2024 13:17:42 +0800 Subject: [PATCH] Update rag_bm.py --- examples/rag_bm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/rag_bm.py b/examples/rag_bm.py index 0ad23c3e4..f604debc6 100644 --- a/examples/rag_bm.py +++ b/examples/rag_bm.py @@ -58,9 +58,9 @@ class RAGExample: logger.info("Loading Exists index!") logger.info(f"Index Path:{output_dir}") self.engine = SimpleEngine.from_index( - index_config=FAISSIndexConfig(), + index_config=FAISSIndexConfig(persist_path=output_dir), ranker_configs=[ColbertRerankConfig()], - retriever_configs=[FAISSRetrieverConfig(persist_path=output_dir), BM25RetrieverConfig()], + retriever_configs=[FAISSRetrieverConfig(), BM25RetrieverConfig()], ) else: logger.info("Loading index from document!")