11 Commits

Author SHA1 Message Date
danqing fu
1378f833bd small fixes 2023-06-08 10:40:15 +08:00
danqing fu
851bb50cb8 small fixes 2023-06-08 09:47:00 +08:00
danqing fu
4d4188aa81 streaming zipformer onnx export 2023-06-07 21:23:11 +08:00
Fangjun Kuang
26f3049f4c minor fixes 2023-06-06 15:30:14 +08:00
Fangjun Kuang
d9e7f02225 Use torch.jit.script() for positional encoding 2023-06-06 15:25:11 +08:00
Fangjun Kuang
4cb2395186 Support longer input for the offline model 2023-06-06 14:20:12 +08:00
danqing fu
09c4946c15 modify max_len to allow longer input audio 2023-06-06 11:09:00 +08:00
danqing fu
8f8b09498f remove if-branch at downsampling pad 2023-06-06 10:05:59 +08:00
Fangjun Kuang
dd91c89f28 Fix exporting non-streaming zipformer to ONNX via torch.jit.trace() 2023-06-02 14:21:28 +08:00
danqing fu
f43c44236f add updated zipformer nonstreaming onnx export 2023-06-01 03:36:11 +00:00
Zengwei Yao
f18b539fbc
Add the upgraded Zipformer model (#1058)
* add the zipformer codes, copied from branch from_dan_scaled_adam_exp1119

* support model export with torch.jit.script

* update RESULTS.md

* support exporting streaming model with torch.jit.script

* add results of streaming models, with some minor changes

* update README.md

* add CI test

* update k2 version in requirements-ci.txt

* update pyproject.toml
2023-05-19 16:47:59 +08:00