extern/whisper.cpp
1
0
Fork 1
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2024-11-08 00:54:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

whisper : handle empty mel (#2324)

Browse Source
This commit is contained in:
Georgi Gerganov 2024-07-27 20:35:04 +03:00 committed by GitHub
parent f68298ce06
commit 6739eb83c3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/whisper.cpp
View File

@ -1920,7 +1920,7 @@ static struct ggml_cgraph * whisper_build_graph_conv(
ggml_set_input(mel_inp);
ggml_tensor * mel;
{
if (ggml_nelements(mel_inp) > 0) {
const int n_len = int(mel_inp->ne[0]);
const int out_s = 2 * n_ctx;
const int i0 = std::min(mel_offset, n_len);
@ -1937,6 +1937,9 @@ static struct ggml_cgraph * whisper_build_graph_conv(
} else {
mel = ggml_cont(ctx0, cur);
}
} else {
// empty mel - just create a dummy tensor with the correct size
mel = ggml_new_tensor_2d(ctx0, GGML_TYPE_F32, 2*n_ctx, n_mels);
}
ggml_set_name(mel, "mel");