add configuable funcname
This commit is contained in:
yichuan520030910320 committed
1 parent
b81b5d0f86
commit
371e3de04e
4 files changed
+973
-1131
No files matched your search
@@ -246,7 +246,7 @@ class DiskannSearcher(LeannBackendSearcherInterface):
|
||||
raise
|
||||
|
||||
def search(self, query: np.ndarray, top_k: int, **kwargs) -> Dict[str, any]:
|
||||
complexity = kwargs.get("complexity", 100)
|
||||
complexity = kwargs.get("complexity", 32)
|
||||
beam_width = kwargs.get("beam_width", 4)
|
||||
|
||||
USE_DEFERRED_FETCH = kwargs.get("USE_DEFERRED_FETCH", False)
|
||||
|
||||