Vector fixes

This commit is contained in:
Louis Dureuil 2024-11-12 16:31:22 +01:00
parent 1fcd5f091e
commit 980921e078
No known key found for this signature in database
5 changed files with 12 additions and 9 deletions

View file

@ -113,7 +113,7 @@ impl Embedder {
threads
.install(move || {
let embeddings: Result<Vec<Vec<Embedding>>, _> = texts
.par_chunks(self.chunk_count_hint())
.par_chunks(self.prompt_count_in_chunk_hint())
.map(move |chunk| self.embed(chunk))
.collect();