Update agent.py
Browse files
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):
|