diff --git a/gpt4all/index/build_index.py b/gpt4all/index/build_index.py index 566f5abd..77a6aa24 100644 --- a/gpt4all/index/build_index.py +++ b/gpt4all/index/build_index.py @@ -89,7 +89,7 @@ def build_index(orig_path, embed_folder_path, index_path): start = 0 while start < len(ds): chunk = ds[start:start + chunk_size] - index.add_items(chunk["embedding"], chunk["index"], num_threads=64) + index.add_items(chunk["embedding"], chunk["id"], num_threads=64) progbar.update(1) start += chunk_size