hiteshwar21 commited on
Commit
c8761c8
·
verified ·
1 Parent(s): 24a35f2

Update config/settings.py

Browse files
Files changed (1) hide show
  1. config/settings.py +4 -3
config/settings.py CHANGED
@@ -10,8 +10,9 @@ class Config:
10
 
11
  # Models - STICK WITH BGE FOR ACCURACY
12
  EMBEDDING_MODEL = "BAAI/bge-large-en-v1.5"
13
- RERANKER_MODEL = "BAAI/bge-reranker-large"
14
-
 
15
  # OpenRouter
16
  OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
17
  OPENROUTER_BASE_URL = "https://openrouter.ai/api/v1"
@@ -53,7 +54,7 @@ class Config:
53
  # return self.PERFORMANCE_MODE in ["quality", "balanced"]
54
  @property
55
  def use_reranker(self):
56
- return False
57
 
58
  @property
59
  def retrieval_k(self):
 
10
 
11
  # Models - STICK WITH BGE FOR ACCURACY
12
  EMBEDDING_MODEL = "BAAI/bge-large-en-v1.5"
13
+ # RERANKER_MODEL = "BAAI/bge-reranker-large"
14
+ RERANKER_MODEL = "cross-encoder/ms-marco-MiniLM-L-6-v2"
15
+
16
  # OpenRouter
17
  OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
18
  OPENROUTER_BASE_URL = "https://openrouter.ai/api/v1"
 
54
  # return self.PERFORMANCE_MODE in ["quality", "balanced"]
55
  @property
56
  def use_reranker(self):
57
+ return True
58
 
59
  @property
60
  def retrieval_k(self):