Commit 9bf3f28
authored
fix linear bias decomposition invokation (#18168)
Fix parakeet/voxtral realtime export using PyTorch release. In both
`export_parakeet_tdt.py` and `export_voxtral_rt.py`, replaced the manual
decomposition dictionary with `torch.export.default_decompositions()`
and added the custom `_linear_bias_decomposition` for
`torch.ops.aten.linear.default`.1 parent 5c1b080 commit 9bf3f28
2 files changed
Lines changed: 6 additions & 8 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
509 | 509 | | |
510 | 510 | | |
511 | 511 | | |
| 512 | + | |
| 513 | + | |
512 | 514 | | |
513 | | - | |
514 | | - | |
515 | 515 | | |
516 | | - | |
517 | | - | |
518 | | - | |
| 516 | + | |
519 | 517 | | |
520 | 518 | | |
521 | 519 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
394 | 394 | | |
395 | 395 | | |
396 | 396 | | |
| 397 | + | |
| 398 | + | |
397 | 399 | | |
398 | | - | |
399 | | - | |
400 | | - | |
| 400 | + | |
401 | 401 | | |
402 | 402 | | |
403 | 403 | | |
| |||
0 commit comments