Whisper AttributeError: 'tuple' object has no attribute 'flatten'

[adriangalilea]

As per the readme suggest I tried:

import whisper

speech_file = "test.mp3"

text = whisper.transcribe(speech_file)["text"]

print(text)

And got

(venv) (base) ➜  whisper git:(main) ✗ python app.py 
Traceback (most recent call last):
  File "/Users/adrian/Developer/mlx-examples/whisper/app.py", line 5, in <module>
    text = whisper.transcribe(speech_file)["text"]
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/adrian/Developer/mlx-examples/whisper/whisper/transcribe.py", line 303, in transcribe
    timestamps = tokens[timestamp_tokens.nonzero().flatten()]
                        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'tuple' object has no attribute 'flatten'

[serinuntius]

I encountered the same error, but resolved it with this PR.

#12

[awni]

Should be fixed with #12