Timothy J. Baek hace 8 meses
padre
commit
bc6f23f82f
Se han modificado 1 ficheros con 3 adiciones y 1 borrados
  1. 3 1
      backend/open_webui/apps/rag/main.py

+ 3 - 1
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(