mirror of
https://github.com/open-webui/open-webui.git
synced 2025-12-15 19:37:47 +01:00
fix
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user