Spaces:
Sleeping
Sleeping
Update retrieval_parent.py
Browse files- retrieval_parent.py +2 -1
retrieval_parent.py
CHANGED
|
@@ -17,7 +17,8 @@ from langchain_core.documents import Document
|
|
| 17 |
|
| 18 |
# --- Configuration ---
|
| 19 |
HYDE_MODEL = "llama-3.1-8b-instant"
|
| 20 |
-
RERANKER_MODEL = '
|
|
|
|
| 21 |
INITIAL_K_CANDIDATES = 20
|
| 22 |
TOP_K_CHUNKS = 10
|
| 23 |
|
|
|
|
| 17 |
|
| 18 |
# --- Configuration ---
|
| 19 |
HYDE_MODEL = "llama-3.1-8b-instant"
|
| 20 |
+
RERANKER_MODEL = 'BAAI/bge-reranker-base'
|
| 21 |
+
#RERANKER_MODEL = 'cross-encoder/ms-marco-MiniLM-L6-v2'
|
| 22 |
INITIAL_K_CANDIDATES = 20
|
| 23 |
TOP_K_CHUNKS = 10
|
| 24 |
|