瀏覽代碼

Update embedding_model.py (#8836)

### What problem does this PR solve?

Remove useless covert for bge encode_queries

### Type of change

- [x] Performance Improvement
tags/v0.20.0
Stephen Hu 3 月之前
父節點
當前提交
5fa6f2f151
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 1 行新增1 行删除
  1. 1
    1
      rag/llm/embedding_model.py

+ 1
- 1
rag/llm/embedding_model.py 查看文件

@@ -114,7 +114,7 @@ class DefaultEmbedding(Base):

def encode_queries(self, text: str):
token_count = num_tokens_from_string(text)
return self._model.encode_queries([text]).tolist()[0], token_count
return self._model.encode_queries([text], convert_to_numpy=False)[0][0].cpu().numpy(), token_count


class OpenAIEmbed(Base):

Loading…
取消
儲存