From 811be520c04ab87fb85a449728164e52c2796cf9 Mon Sep 17 00:00:00 2001 From: Kerollmops Date: Fri, 8 Aug 2025 09:47:08 +0200 Subject: [PATCH] wip: Use Hamming when binary quantized --- crates/milli/src/vector/mod.rs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/crates/milli/src/vector/mod.rs b/crates/milli/src/vector/mod.rs index 50c947f12..ddc5ebffa 100644 --- a/crates/milli/src/vector/mod.rs +++ b/crates/milli/src/vector/mod.rs @@ -4,7 +4,7 @@ use std::sync::{Arc, Mutex}; use std::time::Instant; use deserr::{DeserializeError, Deserr}; -use hannoy::distances::{BinaryQuantizedCosine, Cosine}; +use hannoy::distances::{Cosine, Hamming}; use hannoy::ItemId; use heed::{RoTxn, RwTxn, Unspecified}; use ordered_float::OrderedFloat; @@ -168,7 +168,7 @@ impl VectorStore { // only happens once in the life of an embedder, it's not very performances // sensitive. if quantizing && !self.quantized { - let writer = writer.prepare_changing_distance::(wtxn)?; + let writer = writer.prepare_changing_distance::(wtxn)?; writer .builder(rng) .available_memory(hannoy_memory.unwrap_or(usize::MAX)) @@ -553,7 +553,7 @@ impl VectorStore { self.database.remap_data_type() } - fn quantized_db(&self) -> hannoy::Database { + fn quantized_db(&self) -> hannoy::Database { self.database.remap_data_type() }