From 6739eb83c3ca5cf40d24c6fe8442a761a1eb6248 Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Sat, 27 Jul 2024 20:35:04 +0300 Subject: [PATCH] whisper : handle empty mel (#2324) --- src/whisper.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/whisper.cpp b/src/whisper.cpp index 492a0999e86..752d52901b1 100644 --- a/src/whisper.cpp +++ b/src/whisper.cpp @@ -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");