Fix after upgrading candle

This commit is contained in:
Clément Renault
2025-01-08 15:59:56 +01:00
parent 68333424c6
commit 0ee4671a91
3 changed files with 20 additions and 6 deletions

View File

@@ -175,7 +175,7 @@ impl Embedder {
pub fn embed_tokens(
&self,
tokens: &[usize],
tokens: &[u32],
deadline: Option<Instant>,
) -> Result<Embedding, EmbedError> {
let mut embeddings = embed(&self.data, tokens, 1, Some(self.dimensions), deadline)?;