quantumbit commited on
Commit
cede74c
·
verified ·
1 Parent(s): 6794908

Update config/config.py

Browse files
Files changed (1) hide show
  1. config/config.py +2 -2
config/config.py CHANGED
@@ -49,13 +49,13 @@ ENABLE_HYBRID_SEARCH = True
49
  ENABLE_QUERY_EXPANSION = True
50
  RERANKER_MODEL = "cross-encoder/ms-marco-MiniLM-L-6-v2"
51
  QUERY_EXPANSION_COUNT = 3
52
- MAX_CONTEXT_LENGTH = 4000
53
 
54
  USE_TOTAL_BUDGET_APPROACH = True
55
 
56
  # Embedding Settings
57
  EMBEDDING_MODEL = "BAAI/bge-large-en" # Optimal for CPU
58
- BATCH_SIZE = 8 # Increased for better CPU utilization (adjust based on RAM)
59
 
60
  # Paths
61
  OUTPUT_DIR = os.getenv("RAG_EMBEDDINGS_PATH", "./RAG/rag_embeddings")
 
49
  ENABLE_QUERY_EXPANSION = True
50
  RERANKER_MODEL = "cross-encoder/ms-marco-MiniLM-L-6-v2"
51
  QUERY_EXPANSION_COUNT = 3
52
+ MAX_CONTEXT_LENGTH = 15000
53
 
54
  USE_TOTAL_BUDGET_APPROACH = True
55
 
56
  # Embedding Settings
57
  EMBEDDING_MODEL = "BAAI/bge-large-en" # Optimal for CPU
58
+ BATCH_SIZE = 32 # Increased for better CPU utilization (adjust based on RAM)
59
 
60
  # Paths
61
  OUTPUT_DIR = os.getenv("RAG_EMBEDDINGS_PATH", "./RAG/rag_embeddings")