"vscode:/vscode.git/clone" did not exist on "6affd9cd7ced98fe928293036f24379845e88f9d"
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):
time_duration=time_duration,
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(
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