Fix CrossEncoderRanker (#1531)

Co-authored-by: dong <dongzhancai@iie2.com>
Co-authored-by: Zhancai Dong <dongzhancai1@jd.com>
This commit is contained in:
Cooper 2024-05-20 21:17:06 +08:00 committed by GitHub
parent dcb1bd35d0
commit 04af30e3db
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -239,6 +239,8 @@ class CrossEncoderRanker(Ranker):
Returns:
List[Chunk], reranked candidates
"""
if len(candidates_with_scores) <= 1:
return candidates_with_scores
contents = [candidate.content for candidate in candidates_with_scores]
query_content_pairs = [
[