github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-09-07 19:40:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle edge cases when generating embeddings (#1215)

Browse Source 
* Handle edge cases when generating embeddings
* Improve Python handling & add llmodel_c.h note
- In the Python bindings fail fast with a ValueError when text is empty
- Advice other bindings authors to do likewise in llmodel_c.h
This commit is contained in:
cosmic-snow
2023-07-17 22:21:03 +02:00
committed by GitHub
parent 1e74171a7b
commit 2d02c65177
4 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
gpt4all-backend/llmodel_c.cpp
View File

@@ -168,10 +168,14 @@ void llmodel_prompt(llmodel_model model, const char *prompt,
float *llmodel_embedding(llmodel_model model, const char *text, size_t *embedding_size)
{
if (model == nullptr || text == nullptr || !strlen(text)) {
*embedding_size = 0;
return nullptr;
}
LLModelWrapper *wrapper = reinterpret_cast<LLModelWrapper*>(model);
std::vector<float> embeddingVector = wrapper->llModel->embedding(text);
float *embedding = (float *)malloc(embeddingVector.size() * sizeof(float));
if(embedding == nullptr) {
if (embedding == nullptr) {
*embedding_size = 0;
return nullptr;
}