Timothy J. Baek 7 月之前
父节点
当前提交
bc6f23f82f
共有 1 个文件被更改,包括 3 次插入1 次删除
  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(