Fix after upgrading candle

This commit is contained in:
Clément Renault 2025-01-08 15:59:56 +01:00
parent 68333424c6
commit 0ee4671a91
No known key found for this signature in database
GPG key ID: F250A4C4E3AE5F5F
3 changed files with 20 additions and 6 deletions

View file

@ -1,3 +1,4 @@
use std::fmt;
use std::time::Instant;
use ordered_float::OrderedFloat;
@ -168,7 +169,6 @@ fn infer_api_key() -> String {
.unwrap_or_default()
}
#[derive(Debug)]
pub struct Embedder {
tokenizer: tiktoken_rs::CoreBPE,
rest_embedder: RestEmbedder,
@ -302,3 +302,13 @@ impl Embedder {
self.options.distribution()
}
}
impl fmt::Debug for Embedder {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("Embedder")
.field("tokenizer", &"CoreBPE")
.field("rest_embedder", &self.rest_embedder)
.field("options", &self.options)
.finish()
}
}