summaryrefslogtreecommitdiff
path: root/candle-transformers/src/models/falcon.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-transformers/src/models/falcon.rs')
-rw-r--r--candle-transformers/src/models/falcon.rs2
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-transformers/src/models/falcon.rs b/candle-transformers/src/models/falcon.rs
index 3a3575aa..50ec66f3 100644
--- a/candle-transformers/src/models/falcon.rs
+++ b/candle-transformers/src/models/falcon.rs
@@ -448,7 +448,7 @@ impl Falcon {
vb.pp("transformer.word_embeddings"),
)?;
let blocks = (0..cfg.num_hidden_layers)
- .map(|i| FalconDecoderLayer::load(vb.pp(&format!("transformer.h.{i}")), &cfg))
+ .map(|i| FalconDecoderLayer::load(vb.pp(format!("transformer.h.{i}")), &cfg))
.collect::<Result<Vec<_>>>()?;
let ln_f = layer_norm(
cfg.hidden_size,