Final_Assignment_Template

Running

Coool2 commited on 8 days ago

Commit

517cf5e

verified ·

1 Parent(s): fa1bcdd

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -265,7 +265,7 @@ class DynamicQueryEngineManager:
                 self.visual_reranker = ColPaliRerank(
                     top_n=3,
                     model_name="vidore/colpali-v1.2",
-                    device="cuda"
                 )
             def postprocess_nodes(self, nodes, query_bundle):

                 self.visual_reranker = ColPaliRerank(
                     top_n=3,
                     model_name="vidore/colpali-v1.2",
+                    device = "cpu"
                 )
             def postprocess_nodes(self, nodes, query_bundle):