Coool2 commited on
Commit
bd4ad64
·
1 Parent(s): 999040b

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
agent.py CHANGED
@@ -396,7 +396,8 @@ class DynamicQueryEngineManager:
396
  def __init__(self):
397
  self.text_reranker = SentenceTransformerRerank(
398
  model="cross-encoder/ms-marco-MiniLM-L-2-v2",
399
- top_n=3
 
400
  )
401
 
402
  self.visual_reranker = ColPaliRerank(
 
396
  def __init__(self):
397
  self.text_reranker = SentenceTransformerRerank(
398
  model="cross-encoder/ms-marco-MiniLM-L-2-v2",
399
+ top_n=3,
400
+ device = "cpu"
401
  )
402
 
403
  self.visual_reranker = ColPaliRerank(