江不江
commited on
Commit
·
d3dad46
1
Parent(s):
eec0415
fix embedding_model (#1698)
Browse files### 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 <[email protected]>
rag/llm/embedding_model.py
CHANGED
@@ -128,8 +128,8 @@ class LocalAIEmbed(Base):
|
|
128 |
) # local embedding for LmStudio donot count tokens
|
129 |
|
130 |
def encode_queries(self, text):
|
131 |
-
|
132 |
-
return np.array(
|
133 |
|
134 |
|
135 |
class AzureEmbed(OpenAIEmbed):
|
|
|
128 |
) # local embedding for LmStudio donot count tokens
|
129 |
|
130 |
def encode_queries(self, text):
|
131 |
+
embds, cnt = self.encode([text])
|
132 |
+
return np.array(embds[0]), cnt
|
133 |
|
134 |
|
135 |
class AzureEmbed(OpenAIEmbed):
|