fix diskann for faster mode

This commit is contained in:
yichuan520030910320 committed 2025-08-04 17:46:46 -07:00
1 parent dd71ac8d71
commit efdf6d917a
2 files changed
+3488 -3486

No files matched your search

@@ -312,6 +312,8 @@ class DiskannSearcher(BaseSearcher):
use_global_pruning = True use_global_pruning = True
# Perform search with suppressed C++ output based on log level # 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(): with suppress_cpp_output_if_needed():
labels, distances = self._index.batch_search( labels, distances = self._index.batch_search(
query, query,
@@ -320,9 +322,9 @@ class DiskannSearcher(BaseSearcher):
complexity, complexity,
beam_width, beam_width,
self.num_threads, self.num_threads,
kwargs.get("USE_DEFERRED_FETCH", False), use_deferred_fetch,
kwargs.get("skip_search_reorder", False), kwargs.get("skip_search_reorder", False),
recompute_embeddings, recompute_neighors,
dedup_node_dis, dedup_node_dis,
prune_ratio, prune_ratio,
batch_recompute, batch_recompute,
Generated
+3484 -3484
View File
File diff suppressed because it is too large. Load diff
Reference in new issue
Block a user
Blocking a user prevents them from interacting with repositories, such as opening or commenting on pull requests or issues. Learn more about blocking a user.