瀏覽代碼

Fix: rerank with YoudaoRerank issue. (#6396)

### What problem does this PR solve?

Fix rerank with YoudaoRerank issue,"'YoudaoRerank' object has no
attribute '_dynamic_batch_size'"


![1742541235](https://github.com/user-attachments/assets/9ed304c7-317a-440e-acff-fe895fc20f07)


### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.18.0
zhou 7 月之前
父節點
當前提交
a6aed0da46
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 2 行新增0 行删除
  1. 2
    0
      rag/llm/rerank_model.py

+ 2
- 0
rag/llm/rerank_model.py 查看文件

@@ -194,6 +194,8 @@ class YoudaoRerank(DefaultRerank):
"maidalun1020", "InfiniFlow"))

self._model = YoudaoRerank._model
self._dynamic_batch_size = 8
self._min_batch_size = 1

def similarity(self, query: str, texts: list):
pairs = [(query, truncate(t, self._model.max_length)) for t in texts]

Loading…
取消
儲存