Переглянути джерело

Refactor:remove use less convert for FastEmbed (#8984)

### What problem does this PR solve?

remove use less convert for FastEmbed

### Type of change

- [x] Refactoring
tags/v0.20.0
Stephen Hu 3 місяці тому
джерело
коміт
ec21d9a98f
Аккаунт користувача з таким Email не знайдено
1 змінених файлів з 1 додано та 2 видалено
  1. 1
    2
      rag/llm/embedding_model.py

+ 1
- 2
rag/llm/embedding_model.py Переглянути файл

@@ -352,8 +352,7 @@ class FastEmbed(DefaultEmbedding):
# Using the internal tokenizer to encode the texts and get the total
# number of tokens
encoding = self._model.model.tokenizer.encode(text)
embedding = next(self._model.query_embed(text)).tolist()

embedding = next(self._model.query_embed(text))
return np.array(embedding), len(encoding.ids)



Завантаження…
Відмінити
Зберегти