Wang Baoling commited on
Commit
4cca456
·
1 Parent(s): 8207a08

fix: bug similarity() in YoudaoRerank (#1084)

Browse files

### What problem does this PR solve?

bix fix

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

Files changed (1) hide show
  1. rag/llm/rerank_model.py +1 -1
rag/llm/rerank_model.py CHANGED
@@ -125,7 +125,7 @@ class YoudaoRerank(DefaultRerank):
125
  scores = self._model.compute_score(pairs[i:i + batch_size], max_length=self._model.max_length)
126
  scores = sigmoid(np.array(scores)).tolist()
127
  if isinstance(scores, float): res.append(scores)
128
- res.extend(scores)
129
  return np.array(res), token_count
130
 
131
 
 
125
  scores = self._model.compute_score(pairs[i:i + batch_size], max_length=self._model.max_length)
126
  scores = sigmoid(np.array(scores)).tolist()
127
  if isinstance(scores, float): res.append(scores)
128
+ else: res.extend(scores)
129
  return np.array(res), token_count
130
 
131