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.
This commit is contained in:
kunal-vaishnavi 2023-08-18 11:17:43 -07:00 коммит произвёл GitHub
Родитель 2b50a82171
Коммит 29c6d66c02
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 2 добавлений и 1 удалений

Просмотреть файл

@ -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