AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-12 17:09:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix embedding_model (#1698)

Browse Source 
### What problem does this PR solve?
fix embedding_model #1692
### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

Signed-off-by: seaver <zhudan187@qq.com>
This commit is contained in:
江不江 2024-07-25 11:43:43 +08:00 committed by GitHub
parent b24abee364
commit bc50f68127
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
rag/llm/embedding_model.py
View File

@ -128,8 +128,8 @@ class LocalAIEmbed(Base):
) # local embedding for LmStudio donot count tokens
def encode_queries(self, text):
res = self.client.embeddings.create(text, model=self.model_name)
return np.array(res.data[0].embedding), 1024
embds, cnt = self.encode([text])
return np.array(embds[0]), cnt
class AzureEmbed(OpenAIEmbed):