diff options
author | mert-kurttutan <kurttutan.mert@gmail.com> | 2024-12-24 13:58:21 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-12-24 13:58:21 +0100 |
commit | cd639131f04990c16bfc498ea347cb9df3d2374f (patch) | |
tree | 87d68d1fd0a8e1163e284df62c9b16ec5e976f34 | |
parent | 11aa30be10ebf42d10799a0726a874c74e30ad3e (diff) | |
download | candle-cd639131f04990c16bfc498ea347cb9df3d2374f.tar.gz candle-cd639131f04990c16bfc498ea347cb9df3d2374f.tar.bz2 candle-cd639131f04990c16bfc498ea347cb9df3d2374f.zip |
Fix bug in whisper transformer (#2681)
* Fix bug in whisper transformer
- due to num_threads going to zero
in single threaded case
* Apply rustfmt.
---------
Co-authored-by: Laurent <laurent.mazare@gmail.com>
-rw-r--r-- | candle-transformers/src/models/whisper/audio.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-transformers/src/models/whisper/audio.rs b/candle-transformers/src/models/whisper/audio.rs index 35f9f3df..8490533c 100644 --- a/candle-transformers/src/models/whisper/audio.rs +++ b/candle-transformers/src/models/whisper/audio.rs @@ -204,6 +204,7 @@ pub fn log_mel_spectrogram_<T: Float>( // ensure that the number of threads is even and less than 12 let n_threads = std::cmp::min(get_num_threads() - get_num_threads() % 2, 12); + let n_threads = std::cmp::max(n_threads, 2); let hann = Arc::new(hann); let samples = Arc::new(samples); |