From 734b89570040e97f0c7743c4c0bc28e30a3cd4ee Mon Sep 17 00:00:00 2001 From: polwex Date: Wed, 24 Sep 2025 23:38:36 +0700 Subject: init --- rag/constants.py | 5 +++++ 1 file changed, 5 insertions(+) create mode 100644 rag/constants.py (limited to 'rag/constants.py') diff --git a/rag/constants.py b/rag/constants.py new file mode 100644 index 0000000..327e8e5 --- /dev/null +++ b/rag/constants.py @@ -0,0 +1,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 -- cgit v1.2.3