Quellcode durchsuchen

fix embedding_model (#1698)

### 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>
tags/v0.9.0
江不江 vor 1 Jahr
Ursprung
Commit
bc50f68127
Es ist kein Account mit der E-Mail-Adresse des Committers verbunden
1 geänderte Dateien mit 2 neuen und 2 gelöschten Zeilen
  1. 2
    2
      rag/llm/embedding_model.py

+ 2
- 2
rag/llm/embedding_model.py Datei anzeigen

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

Laden…
Abbrechen
Speichern