fix diskann for faster mode

This commit is contained in:
yichuan520030910320
2025-08-04 17:46:46 -07:00
parent dd71ac8d71
commit efdf6d917a
2 changed files with 3488 additions and 3486 deletions

View File

@@ -312,6 +312,8 @@ class DiskannSearcher(BaseSearcher):
use_global_pruning = True
# Perform search with suppressed C++ output based on log level
use_deferred_fetch = kwargs.get("USE_DEFERRED_FETCH", True)
recompute_neighors = False
with suppress_cpp_output_if_needed():
labels, distances = self._index.batch_search(
query,
@@ -320,9 +322,9 @@ class DiskannSearcher(BaseSearcher):
complexity,
beam_width,
self.num_threads,
kwargs.get("USE_DEFERRED_FETCH", False),
use_deferred_fetch,
kwargs.get("skip_search_reorder", False),
recompute_embeddings,
recompute_neighors,
dedup_node_dis,
prune_ratio,
batch_recompute,

6968
uv.lock generated
View File

File diff suppressed because it is too large Load Diff