diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-09-05 22:46:55 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-09-05 23:46:55 +0200 |
| commit | e3261216b157a7305c18ccdd766b6e2a41afe483 (patch) | |
| tree | edec85bf6faae0c6662e833f74d79ccc2579ef05 /candle-transformers/src/models/distilbert.rs | |
| parent | c02b7c327297f25fc9cc60b8f39c9aa32e40ff21 (diff) | |
| download | candle-e3261216b157a7305c18ccdd766b6e2a41afe483.tar.gz candle-e3261216b157a7305c18ccdd766b6e2a41afe483.tar.bz2 candle-e3261216b157a7305c18ccdd766b6e2a41afe483.zip | |
Clippy fixes for 1.81.0. (#2461)
* Clippy fixes for 1.81.0.
* Another fix.
Diffstat (limited to 'candle-transformers/src/models/distilbert.rs')
| -rw-r--r-- | candle-transformers/src/models/distilbert.rs | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/candle-transformers/src/models/distilbert.rs b/candle-transformers/src/models/distilbert.rs index ea074c97..f899d772 100644 --- a/candle-transformers/src/models/distilbert.rs +++ b/candle-transformers/src/models/distilbert.rs @@ -275,7 +275,7 @@ struct Transformer { impl Transformer { fn load(vb: VarBuilder, config: &Config) -> Result<Self> { let layers = (0..config.n_layers) - .map(|index| TransformerBlock::load(vb.pp(&format!("layer.{index}")), config)) + .map(|index| TransformerBlock::load(vb.pp(format!("layer.{index}")), config)) .collect::<Result<Vec<_>>>()?; let span = tracing::span!(tracing::Level::TRACE, "encoder"); Ok(Transformer { layers, span }) @@ -311,8 +311,8 @@ impl DistilBertModel { (Err(err), _) | (_, Err(err)) => { if let Some(model_type) = &config.model_type { if let (Ok(embeddings), Ok(encoder)) = ( - Embeddings::load(vb.pp(&format!("{model_type}.embeddings")), config), - Transformer::load(vb.pp(&format!("{model_type}.transformer")), config), + Embeddings::load(vb.pp(format!("{model_type}.embeddings")), config), + Transformer::load(vb.pp(format!("{model_type}.transformer")), config), ) { (embeddings, encoder) } else { |