rag/constants.py


1
2
3
4
5

EMBED_MODEL_ID = "Qwen/Qwen3-Embedding-8B"
RERANKER_ID = "Qwen/Qwen3-Reranker-8B"  # or -large if you’ve got VRAM
# RERANKER_ID = "BAAI/bge-reranker-base"  # or -large if you’ve got VRAM
MAX_TOKENS = 600
BATCH = 16