diff --git a/candle-transformers/src/models/bert.rs b/candle-transformers/src/models/bert.rs index 1df2683574..bdc0385deb 100644 --- a/candle-transformers/src/models/bert.rs +++ b/candle-transformers/src/models/bert.rs @@ -554,8 +554,7 @@ impl BertLMPredictionHead { impl Module for BertLMPredictionHead { fn forward(&self, hidden_states: &Tensor) -> Result { - self - .decoder + self.decoder .forward(&self.transform.forward(hidden_states)?) } }