diff options
Diffstat (limited to 'candle-examples/examples/whisper/main.rs')
| -rw-r--r-- | candle-examples/examples/whisper/main.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-examples/examples/whisper/main.rs b/candle-examples/examples/whisper/main.rs index d7b303cf..079424e3 100644 --- a/candle-examples/examples/whisper/main.rs +++ b/candle-examples/examples/whisper/main.rs @@ -127,7 +127,7 @@ impl Decoder { .to_scalar::<f32>()? as f64; } - let (seq_len, _) = logits.shape().r2()?; + let (seq_len, _) = logits.dims2()?; let logits = logits .get(seq_len - 1)? .broadcast_add(&self.suppress_tokens)?; @@ -195,7 +195,7 @@ impl Decoder { } fn run(&mut self, mel: &Tensor) -> Result<Vec<Segment>> { - let (_, _, content_frames) = mel.shape().r3()?; + let (_, _, content_frames) = mel.dims3()?; let mut seek = 0; let mut segments = vec![]; while seek < content_frames { |