From 29c6d66c02d96f3c003a7083f6fe0096ace90af8 Mon Sep 17 00:00:00 2001 From: kunal-vaishnavi <115581922+kunal-vaishnavi@users.noreply.github.com> Date: Fri, 18 Aug 2023 11:17:43 -0700 Subject: [PATCH] Separate variable definition in Whisper E2E script (#535) This PR fixes running this script by defining the number of mels and number of frames on separate lines. --- tutorials/whisper_e2e.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tutorials/whisper_e2e.py b/tutorials/whisper_e2e.py index 07779e9d..dca2156d 100644 --- a/tutorials/whisper_e2e.py +++ b/tutorials/whisper_e2e.py @@ -74,7 +74,8 @@ def get_model_inputs(ort_session, audio_data): # Add optional inputs if present in model batch_size = 1 - N_MELS = 80, N_FRAMES = 3000 + N_MELS = 80 + N_FRAMES = 3000 vocab_size = 51864 if ".en" in MODEL_NAME else 51865 decoder_start_token_id = 50257 if ".en" in MODEL_NAME else 50258