zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-12-05 02:58:13 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

work with tuple shape (#393)

Browse Source
This commit is contained in:
Awni Hannun
2024-02-01 13:03:47 -08:00
committed by GitHub
parent 0340113e02
commit ec14583c2a
6 changed files with 5 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
whisper/whisper/decoding.py
View File

@@ -50,7 +50,7 @@ def detect_language(
mel = mel[None]
# skip encoder forward pass if already-encoded audio features were given
if mel.shape[-2:] != [model.dims.n_audio_ctx, model.dims.n_audio_state]:
if mel.shape[-2:] != (model.dims.n_audio_ctx, model.dims.n_audio_state):
mel = model.encoder(mel)
# forward pass using a single token, startoftranscript