diff options
Diffstat (limited to 'candle-wasm-example/src')
| -rw-r--r-- | candle-wasm-example/src/model.rs | 6 | ||||
| -rw-r--r-- | candle-wasm-example/src/worker.rs | 4 |
2 files changed, 5 insertions, 5 deletions
diff --git a/candle-wasm-example/src/model.rs b/candle-wasm-example/src/model.rs index 89c0d708..97eff839 100644 --- a/candle-wasm-example/src/model.rs +++ b/candle-wasm-example/src/model.rs @@ -164,7 +164,7 @@ impl MultiHeadAttention { } fn reshape_head(&self, x: &Tensor) -> Result<Tensor> { - let (n_batch, n_ctx, n_state) = x.shape().r3()?; + let (n_batch, n_ctx, n_state) = x.dims3()?; let target_dims = &[n_batch, n_ctx, self.n_head, n_state / self.n_head]; Ok(x.reshape(target_dims)?.transpose(1, 2)?) } @@ -176,7 +176,7 @@ impl MultiHeadAttention { v: &Tensor, mask: Option<&Tensor>, ) -> Result<Tensor> { - let (_, n_ctx, n_state) = q.shape().r3()?; + let (_, n_ctx, n_state) = q.dims3()?; let scale = ((n_state / self.n_head) as f64).powf(-0.25); let q = { let _timer = crate::Timer::new("q::reshape"); @@ -328,7 +328,7 @@ impl AudioEncoder { self.conv2.forward(&x)?.gelu()? }; let x = x.transpose(1, 2)?; - let (_bsize, seq_len, _hidden) = x.shape().r3()?; + let (_bsize, seq_len, _hidden) = x.dims3()?; let positional_embedding = self.positional_embedding.narrow(0, 0, seq_len)?; let mut x = x.broadcast_add(&positional_embedding)?; for block in self.blocks.iter() { diff --git a/candle-wasm-example/src/worker.rs b/candle-wasm-example/src/worker.rs index 5001e7e4..ea64bf02 100644 --- a/candle-wasm-example/src/worker.rs +++ b/candle-wasm-example/src/worker.rs @@ -134,7 +134,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)?; @@ -207,7 +207,7 @@ impl Decoder { fn run(&self, mel: &Tensor) -> anyhow::Result<Vec<Segment>> { let mut rng = StdRng::seed_from_u64(299792458); - let (_, _, content_frames) = mel.shape().r3()?; + let (_, _, content_frames) = mel.dims3()?; let mut seek = 0; let mut segments = vec![]; while seek < content_frames { |