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(