diff options
Diffstat (limited to 'candle-examples/examples/whisper/model.rs')
| -rw-r--r-- | candle-examples/examples/whisper/model.rs | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/candle-examples/examples/whisper/model.rs b/candle-examples/examples/whisper/model.rs index 9f0c9ef8..d6bea09a 100644 --- a/candle-examples/examples/whisper/model.rs +++ b/candle-examples/examples/whisper/model.rs @@ -281,11 +281,13 @@ impl AudioEncoder { padding: 1, stride: 1, groups: 1, + dilation: 1, }; let cfg2 = Conv1dConfig { padding: 1, stride: 2, groups: 1, + dilation: 1, }; let conv1 = conv1d(cfg.num_mel_bins, n_state, 3, cfg1, vb.pp("conv1"))?; let conv2 = conv1d(n_state, n_state, 3, cfg2, vb.pp("conv2"))?; |