huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rename the .r functions to .dims so as to be a bit more explicit. (#220)

Browse Source
This commit is contained in:
Laurent Mazare
2023-07-22 11:39:27 +02:00
committed by GitHub
parent 52c5d8c087
commit 43c7223292
18 changed files with 56 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-examples/examples/whisper/main.rs
View File

@ -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 {