UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-18 16:51:17 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Timothy J. Baek 2024-09-16 12:33:55 +02:00
parent cb9e76c7f9
commit bc6f23f82f
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/open_webui/apps/rag/main.py
View File

@ -206,7 +206,9 @@ def update_reranking_model(
self.ckpt = Checkpoint(name, colbert_config=ColBERTConfig()) self.ckpt = Checkpoint(name, colbert_config=ColBERTConfig())
pass pass
def calculate_similarity_scores(query_embeddings, document_embeddings): def calculate_similarity_scores(
self, query_embeddings, document_embeddings
):
# Validate dimensions to ensure compatibility # Validate dimensions to ensure compatibility
if query_embeddings.dim() != 3: if query_embeddings.dim() != 3:
raise ValueError( raise ValueError(