osmarks/meme-search-engine
1
0
Fork 0
mirror of https://github.com/osmarks/meme-search-engine.git synced 2025-02-23 14:30:04 +00:00
Code Issues Releases Wiki Activity

add missing files

Browse Source
This commit is contained in:
osmarks 2025-01-18 17:09:44 +00:00
parent a5a6e960bb
commit fcd28a5ede
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
diskann/src/vector.rs
View File

@ -316,7 +316,7 @@ pub struct ProductQuantizer {
} }
// chunk * centroid_index // chunk * centroid_index
pub struct DistanceLUT(Vec<f32>); pub struct QueryLUT(Vec<f32>);
impl ProductQuantizer { impl ProductQuantizer {
pub fn apply_transform(&self, x: &[f32]) -> Vec<f32> { pub fn apply_transform(&self, x: &[f32]) -> Vec<f32> {
@ -366,7 +366,7 @@ impl ProductQuantizer {
} }
// not particularly performance-sensitive right now; do unbatched // not particularly performance-sensitive right now; do unbatched
pub fn preprocess_query(&self, query: &[f32]) -> DistanceLUT { pub fn preprocess_query(&self, query: &[f32]) -> QueryLUT {
let transformed = self.apply_transform(query); let transformed = self.apply_transform(query);
let n_chunks = self.n_dims / self.n_dims_per_code; let n_chunks = self.n_dims / self.n_dims_per_code;
let n_centroids = self.centroids.len() / self.n_dims; let n_centroids = self.centroids.len() / self.n_dims;
@ -382,11 +382,11 @@ impl ProductQuantizer {
} }
} }
DistanceLUT(lut) QueryLUT(lut)
} }
// compute dot products of query against product-quantized vectors // compute dot products of query against product-quantized vectors
pub fn asymmetric_dot_product(&self, query: &DistanceLUT, pq_vectors: &[u8]) -> Vec<i64> { pub fn asymmetric_dot_product(&self, query: &QueryLUT, pq_vectors: &[u8]) -> Vec<i64> {
let n_chunks = self.n_dims / self.n_dims_per_code; let n_chunks = self.n_dims / self.n_dims_per_code;
let n_vectors = pq_vectors.len() / n_chunks; let n_vectors = pq_vectors.len() / n_chunks;
let mut scores = vec![0.0; n_vectors]; let mut scores = vec![0.0; n_vectors];

8
frontend_config.json
View File

@ -1,5 +1,5 @@
{ {
"backend_url": "http://localhost:1707/", "backend_url": "http://localhost:5601/",
"image_path": "http://localhost:7858/", "image_path": "",
"thumb_path": "http://localhost:7857/" "thumb_path": null
} }