diff options
| author | polwex <polwex@sortug.com> | 2025-09-24 23:38:36 +0700 |
|---|---|---|
| committer | polwex <polwex@sortug.com> | 2025-09-24 23:38:36 +0700 |
| commit | 734b89570040e97f0c7743c4c0bc28e30a3cd4ee (patch) | |
| tree | 7142d9f37908138c38d0ade066e960c3a1c69f5d /rag/constants.py | |
| parent | 57aaafdb137fe49930711f6ed5ccc83b3a119cd2 (diff) | |
Diffstat (limited to 'rag/constants.py')
| -rw-r--r-- | rag/constants.py | 5 |
1 files changed, 5 insertions, 0 deletions
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 |