From 9a1e54ebd4137c2857adf1818d37b5f9bae6fecd Mon Sep 17 00:00:00 2001 From: csunny Date: Thu, 18 May 2023 23:31:15 +0800 Subject: [PATCH] top_k update --- pilot/configs/model_config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pilot/configs/model_config.py b/pilot/configs/model_config.py index 2a4f5eac4..5ab2e4fbe 100644 --- a/pilot/configs/model_config.py +++ b/pilot/configs/model_config.py @@ -29,6 +29,6 @@ ISLOAD_8BIT = True ISDEBUG = False -VECTOR_SEARCH_TOP_K = 3 +VECTOR_SEARCH_TOP_K = 10 VS_ROOT_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)), "vs_store") KNOWLEDGE_UPLOAD_ROOT_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)), "data") \ No newline at end of file