From 057ed6f6114af61267e9f79695fbdef30200c3c1 Mon Sep 17 00:00:00 2001 From: Athe-kunal Date: Sat, 28 Sep 2024 02:40:24 -0500 Subject: [PATCH] merge changes from main --- byaldi/colpali.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/byaldi/colpali.py b/byaldi/colpali.py index fd3967a..1cbf44c 100644 --- 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()