summaryrefslogtreecommitdiff
path: root/candle-transformers/src/models/segformer.rs
diff options
context:
space:
mode:
authorLaurent Mazare <laurent.mazare@gmail.com>2024-09-05 22:46:55 +0100
committerGitHub <noreply@github.com>2024-09-05 23:46:55 +0200
commite3261216b157a7305c18ccdd766b6e2a41afe483 (patch)
treeedec85bf6faae0c6662e833f74d79ccc2579ef05 /candle-transformers/src/models/segformer.rs
parentc02b7c327297f25fc9cc60b8f39c9aa32e40ff21 (diff)
downloadcandle-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/segformer.rs')
-rw-r--r--candle-transformers/src/models/segformer.rs8
1 files changed, 4 insertions, 4 deletions
diff --git a/candle-transformers/src/models/segformer.rs b/candle-transformers/src/models/segformer.rs
index 3727e004..260ceb3a 100644
--- a/candle-transformers/src/models/segformer.rs
+++ b/candle-transformers/src/models/segformer.rs
@@ -404,7 +404,7 @@ impl SegformerEncoder {
stride,
num_channels,
hidden_size,
- vb.pp(&format!("patch_embeddings.{}", i)),
+ vb.pp(format!("patch_embeddings.{}", i)),
)?);
let mut layers = Vec::with_capacity(config.depths[i]);
for j in 0..config.depths[i] {
@@ -417,14 +417,14 @@ impl SegformerEncoder {
num_attention_heads,
sequence_reduction_ratio,
mlp_ratio,
- vb.pp(&format!("block.{}.{}", i, j)),
+ vb.pp(format!("block.{}.{}", i, j)),
)?);
}
blocks.push(layers);
layer_norms.push(layer_norm(
hidden_size,
config.layer_norm_eps,
- vb.pp(&format!("layer_norm.{}", i)),
+ vb.pp(format!("layer_norm.{}", i)),
)?);
}
Ok(Self {
@@ -507,7 +507,7 @@ impl SegformerDecodeHead {
linear_c.push(SegformerMLP::new(
config,
hidden_size,
- vb.pp(&format!("linear_c.{}", i)),
+ vb.pp(format!("linear_c.{}", i)),
)?);
}
let linear_fuse = conv2d_no_bias(