Skip to content

Commit efdf6d9

Browse files
committed
fix diskann for faster mode
1 parent dd71ac8 commit efdf6d9

2 files changed

Lines changed: 3535 additions & 3533 deletions

File tree

packages/leann-backend-diskann/leann_backend_diskann/diskann_backend.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -312,6 +312,8 @@ def search(
312312
use_global_pruning = True
313313

314314
# Perform search with suppressed C++ output based on log level
315+
use_deferred_fetch = kwargs.get("USE_DEFERRED_FETCH", True)
316+
recompute_neighors = False
315317
with suppress_cpp_output_if_needed():
316318
labels, distances = self._index.batch_search(
317319
query,
@@ -320,9 +322,9 @@ def search(
320322
complexity,
321323
beam_width,
322324
self.num_threads,
323-
kwargs.get("USE_DEFERRED_FETCH", False),
325+
use_deferred_fetch,
324326
kwargs.get("skip_search_reorder", False),
325-
recompute_embeddings,
327+
recompute_neighors,
326328
dedup_node_dis,
327329
prune_ratio,
328330
batch_recompute,

0 commit comments

Comments
 (0)