merge changes from main

AnswerDotAI · Sep 28, 2024 · 057ed6f · 057ed6f
1 parent 0e11a7f
commit 057ed6f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/byaldi/colpali.py b/byaldi/colpali.py
@@ -618,7 +618,7 @@ def search(
             else:
                 req_embeddings, req_embedding_ids = self.filter_embeddings(filter_metadata=filter_metadata) 
             # Compute scores
-            scores = self.processor.score(qs,req_embeddings, self.indexed_embeddings).cpu().numpy()
+            scores = self.processor.score(qs,req_embeddings).cpu().numpy()
 
             # Get top k relevant pages
             top_pages = scores.argsort(axis=1)[0][-k:][::-1].tolist()