-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathconfig.py
More file actions
17 lines (13 loc) · 736 Bytes
/
Copy pathconfig.py
File metadata and controls
17 lines (13 loc) · 736 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from pathlib import Path
REPO_ROOT = Path(__file__).parent.parent
PAPERS: dict[str, dict] = {
"ocr__2605.06652v1__9d24d9cf929e3ec3": {
"ocr_json": REPO_ROOT / "RAG_research_paper" / "data" / "normalized" / "ocr__2605.06652v1__9d24d9cf929e3ec3.json",
"chunks_json": REPO_ROOT / "RAG_research_paper" / "data" / "chunks" / "ocr__2605.06652v1__9d24d9cf929e3ec3.json",
"images_dir": REPO_ROOT / "research_paper_ocr" / "data_ocr" / "mineru_raw" / "ocr__2605.06652v1__9d24d9cf929e3ec3" / "2605.06652v1" / "auto",
}
}
TOP_N_RESULTS = 10
RETRIEVAL_BACKEND = "cross_encoder" # "bi_encoder" | "cross_encoder"
CROSS_ENCODER_MODEL = "Qwen/Qwen3-Reranker-0.6B"
BI_ENCODER_MODEL = "microsoft/harrier-oss-v1-0.6b"