diff --git a/milli/src/vector/hf.rs b/milli/src/vector/hf.rs index 7acb09aa8..cdfdbfb75 100644 --- a/milli/src/vector/hf.rs +++ b/milli/src/vector/hf.rs @@ -151,7 +151,8 @@ impl Embedder { let token_ids = tokens .iter() .map(|tokens| { - let tokens = tokens.get_ids().to_vec(); + let mut tokens = tokens.get_ids().to_vec(); + tokens.truncate(512); Tensor::new(tokens.as_slice(), &self.model.device).map_err(EmbedError::tensor_shape) }) .collect::, EmbedError>>()?;