kvaishnavi commited on
Commit
a0ecd43
·
verified ·
1 Parent(s): 05f620b

Update onnx/builder.py

Browse files
Files changed (1) hide show
  1. onnx/builder.py +1 -1
onnx/builder.py CHANGED
@@ -127,7 +127,7 @@ def build_speech(args):
127
  dummy_inputs = (
128
  inputs["input_audio_embeds"], # audio_embeds: torch.FloatTensor
129
  inputs["audio_attention_mask"], # audio_attention_mask: torch.BoolTensor
130
- inputs["audio_embed_sizes"], # audio_sizes: torch.FloatTensor
131
  inputs["input_mode"], # audio_projection_mode: int
132
  )
133
  dynamic_axes = {
 
127
  dummy_inputs = (
128
  inputs["input_audio_embeds"], # audio_embeds: torch.FloatTensor
129
  inputs["audio_attention_mask"], # audio_attention_mask: torch.BoolTensor
130
+ inputs["audio_embed_sizes"], # audio_sizes: torch.LongTensor
131
  inputs["input_mode"], # audio_projection_mode: int
132
  )
133
  dynamic_axes = {