ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-15 13:55:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

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(