diff --git a/xinference/model/rerank/core.py b/xinference/model/rerank/core.py index 1f8c8e050d..de9a7d33ae 100644 --- a/xinference/model/rerank/core.py +++ b/xinference/model/rerank/core.py @@ -268,6 +268,12 @@ def rerank( similarity_scores = self._model.compute_score(sentence_combinations) if not isinstance(similarity_scores, Sequence): similarity_scores = [similarity_scores] + elif ( + isinstance(similarity_scores, list) + and len(similarity_scores) > 0 + and isinstance(similarity_scores[0], Sequence) + ): + similarity_scores = similarity_scores[0] sim_scores_argsort = list(reversed(np.argsort(similarity_scores))) if top_n is not None: