Coool2 commited on
Commit
517cf5e
·
verified ·
1 Parent(s): fa1bcdd

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
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="cuda"
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):