Skip to content

Commit 0fd8114

Browse files
Fix crash when creating tensor from empty buffer in whisper (#454)
Co-authored-by: Jonatan Kłosko <jonatanklosko@gmail.com>
1 parent 273805e commit 0fd8114

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

lib/bumblebee/audio/speech_to_text_whisper.ex

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -318,8 +318,12 @@ defmodule Bumblebee.Audio.SpeechToTextWhisper do
318318
buffer = buffer ++ [chunk]
319319
full_chunks([], {buffer, buffer_size}, chunk_length, step)
320320
end,
321-
fn {buffer, buffer_size} ->
322-
{[Nx.concatenate(buffer)], {buffer, buffer_size}}
321+
fn
322+
{[], buffer_size} ->
323+
{[], {[], buffer_size}}
324+
325+
{buffer, buffer_size} ->
326+
{[Nx.concatenate(buffer)], {buffer, buffer_size}}
323327
end,
324328
fn _ -> :ok end
325329
)

0 commit comments

Comments
 (0)