Update agent.py
Browse files
agent.py
CHANGED
@@ -265,7 +265,7 @@ class DynamicQueryEngineManager:
|
|
265 |
self.visual_reranker = ColPaliRerank(
|
266 |
top_n=3,
|
267 |
model_name="vidore/colpali-v1.2",
|
268 |
-
device="
|
269 |
)
|
270 |
|
271 |
def postprocess_nodes(self, nodes, query_bundle):
|
|
|
265 |
self.visual_reranker = ColPaliRerank(
|
266 |
top_n=3,
|
267 |
model_name="vidore/colpali-v1.2",
|
268 |
+
device = "cpu"
|
269 |
)
|
270 |
|
271 |
def postprocess_nodes(self, nodes, query_bundle):
|