Update embedding_model.py (#8836)

### What problem does this PR solve?

Remove useless covert for bge encode_queries

### Type of change

- [x] Performance Improvement
This commit is contained in:
Stephen Hu 2025-07-15 14:04:58 +08:00 committed by GitHub
parent 51a8604dcb
commit 5fa6f2f151
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

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