Bläddra i källkod

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 månader sedan
förälder
incheckning
ec21d9a98f
Inget konto är kopplat till bidragsgivarens mejladress
1 ändrade filer med 1 tillägg och 2 borttagningar
  1. 1
    2
      rag/llm/embedding_model.py

+ 1
- 2
rag/llm/embedding_model.py Visa fil

@@ -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)



Laddar…
Avbryt
Spara