riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 18:51:14 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix whisper export (#20800)

Browse Source 
* fix_whisper_export

* update input

* update input
This commit is contained in:
Mohit Sharma 2022-12-21 20:58:42 +05:30 committed by GitHub
parent 3090e70857
commit aaa6296de2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/models/whisper/configuration_whisper.py
View File

@ -264,6 +264,9 @@ class WhisperOnnxConfig(OnnxSeq2SeqConfigWithPast):
time_duration=time_duration, time_duration=time_duration,
frequency=frequency, frequency=frequency,
) )
encoder_sequence_length = encoder_inputs["input_features"].shape[2]
seq_length = encoder_sequence_length // 2 if self.use_past else seq_length
decoder_inputs = super().generate_dummy_inputs( decoder_inputs = super().generate_dummy_inputs(
preprocessor.tokenizer, batch_size, seq_length, is_pair, framework preprocessor.tokenizer, batch_size, seq_length, is_pair, framework
) )