Unverified Commit aaa6296d authored by Mohit Sharma's avatar Mohit Sharma Committed by GitHub
Browse files

Fix whisper export (#20800)

* fix_whisper_export

* update input

* update input
parent 3090e708
...@@ -264,6 +264,9 @@ class WhisperOnnxConfig(OnnxSeq2SeqConfigWithPast): ...@@ -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
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment