9 Commits

Author SHA1 Message Date
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
7b00b34617 minor fixes 2023-06-02 14:23:47 +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
Fangjun Kuang
7b0afbdc16
Remove cur_batch_idx (#1102) 2023-05-30 14:49:54 +08:00
Zengwei Yao
6826b076d4
add flops profiler, support for Zipformer encoder and Conformer encoder (#1093)
* add flops profiler, support for Zipformer encoder and Conformer encoder

* support for reworked conformer and old zipformer

* skip black check
2023-05-24 19:10:45 +08:00
Zengwei Yao
30fcd16c7d
rm zipformer/__init__.py (#1075) 2023-05-20 23:12:11 +08: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