Coool2 commited on
Commit
da7e61a
·
verified ·
1 Parent(s): b349aa6

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -0
agent.py CHANGED
@@ -265,6 +265,8 @@ class DynamicQueryEngineManager:
265
  )
266
  self.visual_reranker = ColPaliRerank(
267
  top_n=3,
 
 
268
  device="cpu"
269
  )
270
  def postprocess_nodes(self, nodes, query_bundle):
 
265
  )
266
  self.visual_reranker = ColPaliRerank(
267
  top_n=3,
268
+ model="vidore/colpali-v1.2",
269
+ keep_retrieval_score=True,
270
  device="cpu"
271
  )
272
  def postprocess_nodes(self, nodes, query_bundle):