Closed John-Goff closed 6 months ago
distil-whisper/distil-small.en
and related models also fails with the same error.
For these you need to set task: nil
explicitly. Bumblebee main has an improved error message, it just hasn't been released yet :)
the generation config does not have any tasks defined. If you are dealing with a monolingual model, set :task to nil. Otherwise you may need to update generation_config.extra_config.task_to_token_id
For the record, discussed previously in https://github.com/elixir-nx/bumblebee/issues/267.
With any combination of
whisper-medium.en
,whisper-small.en
, orwhisper-tiny.en
the following code does not work:Fails with error
Changing the model to
whisper-medium
,whisper-small
, orwhisper-tiny
will work properly.