Skip to content

Commit

Permalink
Use b1 quantization for USearch
Browse files Browse the repository at this point in the history
As suggested by @adolfogc (in unum-cloud/usearch#405)
the USearch implementation should use `b1` instead of `i8`
for binary representations.
  • Loading branch information
ashvardanian committed May 14, 2024
1 parent de3b298 commit 8d17c89
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion sentence_transformers/quantization.py
Original file line number Diff line number Diff line change
Expand Up @@ -239,7 +239,7 @@ def semantic_search_usearch(
corpus_index = Index(
ndim=corpus_embeddings.shape[1],
metric="hamming",
dtype="i8",
dtype="b1",
)
corpus_index.add(np.arange(len(corpus_embeddings)), corpus_embeddings)

Expand Down

0 comments on commit 8d17c89

Please sign in to comment.