add configuable funcname

This commit is contained in:
yichuan520030910320 committed 2025-07-02 05:18:00 +00:00
1 parent 371e3de04e
commit 44815ee7fd
5 files changed
+3208 -966

No files matched your search

@@ -192,6 +192,7 @@ class HNSWBuilder(LeannBackendBuilderInterface):
build_kwargs = {**self.build_params, **kwargs}
metric_str = build_kwargs.get("distance_metric", "mips").lower()
metric_enum = get_metric_map().get(metric_str)
print('metric_enum', metric_enum,' metric_str', metric_str)
if metric_enum is None:
raise ValueError(f"Unsupported distance_metric '{metric_str}'.")
@@ -217,7 +218,9 @@ class HNSWBuilder(LeannBackendBuilderInterface):
faiss.normalize_L2(data)
# Add vectors to index
print('starting to add vectors to index')
index.add(data.shape[0], faiss.swig_ptr(data))
print('vectors added to index')
# Save index
index_file = index_dir / f"{index_prefix}.index"
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.