mirror of
https://github.com/open-webui/open-webui.git
synced 2025-12-13 12:55:19 +00:00
fix
This commit is contained in:
parent
cb9e76c7f9
commit
bc6f23f82f
1 changed files with 3 additions and 1 deletions
|
|
@ -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(
|
||||
|
|
|
|||
Loading…
Reference in a new issue