From bc6f23f82f5bf60161970562fe90486159086946 Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Mon, 16 Sep 2024 12:33:55 +0200 Subject: [PATCH] fix --- backend/open_webui/apps/rag/main.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/backend/open_webui/apps/rag/main.py b/backend/open_webui/apps/rag/main.py index f801d7423..eab197273 100644 --- a/backend/open_webui/apps/rag/main.py +++ b/backend/open_webui/apps/rag/main.py @@ -206,7 +206,9 @@ def update_reranking_model( self.ckpt = Checkpoint(name, colbert_config=ColBERTConfig()) pass - def calculate_similarity_scores(query_embeddings, document_embeddings): + def calculate_similarity_scores( + self, query_embeddings, document_embeddings + ): # Validate dimensions to ensure compatibility if query_embeddings.dim() != 3: raise ValueError(