fix: index -> id

This commit is contained in:
Zach Nussbaum 2023-04-25 20:33:49 +00:00
parent 869829a065
commit 7832707c37

View File

@ -89,7 +89,7 @@ def build_index(orig_path, embed_folder_path, index_path):
start = 0 start = 0
while start < len(ds): while start < len(ds):
chunk = ds[start:start + chunk_size] 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) progbar.update(1)
start += chunk_size start += chunk_size