Final_Assignment_Template

Configuration error

oremaz commited on Jun 1

Commit

543ac1a

1 Parent(s): d37c3d6

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):