Explorar el Código

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 hace 3 meses
padre
commit
ec21d9a98f
No account linked to committer's email address
Se han modificado 1 ficheros con 1 adiciones y 2 borrados
  1. 1
    2
      rag/llm/embedding_model.py

+ 1
- 2
rag/llm/embedding_model.py Ver fichero

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



Cargando…
Cancelar
Guardar