Update onnx/builder.py
Browse files- 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.
|
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 = {
|