NLP_Lab / src /config.py
apytel
Switches data directory to local path for easier access
5582d21
Raw
History Blame Contribute Delete
652 Bytes
EMBED_MODEL = "BAAI/bge-small-en-v1.5"
RERANK_MODEL = "BAAI/bge-reranker-base"
CHUNK_SIZE = 700 # tokens
CHUNK_OVERLAP = 120
TOP_K_BM25 = 20
TOP_K_DENSE = 20
TOP_K_FUSED = 30
TOP_N_FINAL = 5
RRF_K = 60
DEFAULT_MODEL = "Qwen/Qwen2.5-Coder-32B-Instruct"
HF_MODELS = [
"Qwen/Qwen2.5-Coder-32B-Instruct",
"meta-llama/Llama-3.3-70B-Instruct",
"meta-llama/Llama-3.1-8B-Instruct",
"mistralai/Mistral-7B-Instruct-v0.3",
]
DATA_DIR = "data"
CHUNKS_FILE = DATA_DIR + "/chunks.parquet"
FAISS_FILE = DATA_DIR + "/index.faiss"
BM25_FILE = DATA_DIR + "/bm25.pkl"
WIKI_BASE_URL = "https://wiki.freecad.org"