diff options
Diffstat (limited to 'candle-transformers/src/models/jina_bert.rs')
| -rw-r--r-- | candle-transformers/src/models/jina_bert.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-transformers/src/models/jina_bert.rs b/candle-transformers/src/models/jina_bert.rs index a9ae37e9..1f0fae1e 100644 --- a/candle-transformers/src/models/jina_bert.rs +++ b/candle-transformers/src/models/jina_bert.rs @@ -344,7 +344,7 @@ impl BertEncoder { candle::bail!("only alibi is supported as a position-embedding-type") } let layers = (0..cfg.num_hidden_layers) - .map(|index| BertLayer::new(vb.pp(&format!("layer.{index}")), cfg)) + .map(|index| BertLayer::new(vb.pp(format!("layer.{index}")), cfg)) .collect::<Result<Vec<_>>>()?; let span = tracing::span!(tracing::Level::TRACE, "encoder"); let alibi = build_alibi_bias(cfg)?.to_device(vb.device())?; |