File tree Expand file tree Collapse file tree 2 files changed +34
-0
lines changed Expand file tree Collapse file tree 2 files changed +34
-0
lines changed Original file line number Diff line number Diff line change @@ -89,6 +89,29 @@ def get_models():
8989 popd
9090 """ ,
9191 ),
92+ Model (
93+ model_name = "sherpa-onnx-paraformer-zh-small-2024-03-09" ,
94+ idx = 14 ,
95+ lang = "zh" ,
96+ short_name = "small_paraformer" ,
97+ rule_fsts = "itn_zh_number.fst" ,
98+ cmd = """
99+ if [ ! -f itn_zh_number.fst ]; then
100+ curl -SL -O https://github.com/k2-fsa/sherpa-onnx/releases/download/asr-models/itn_zh_number.fst
101+ fi
102+ pushd $model_name
103+
104+ rm -fv README.md
105+ rm -fv *.py
106+ rm -fv *.yaml
107+ rm -fv *.mvn
108+ rm -rfv test_wavs
109+
110+ ls -lh
111+
112+ popd
113+ """ ,
114+ ),
92115 Model (
93116 model_name = "icefall-asr-zipformer-wenetspeech-20230615" ,
94117 idx = 4 ,
Original file line number Diff line number Diff line change @@ -310,6 +310,17 @@ fun getOfflineModelConfig(type: Int): OfflineModelConfig? {
310310 modelType = " transducer" ,
311311 )
312312 }
313+
314+ 14 -> {
315+ val modelDir = " sherpa-onnx-paraformer-zh-small-2024-03-09"
316+ return OfflineModelConfig (
317+ paraformer = OfflineParaformerModelConfig (
318+ model = " $modelDir /model.int8.onnx" ,
319+ ),
320+ tokens = " $modelDir /tokens.txt" ,
321+ modelType = " paraformer" ,
322+ )
323+ }
313324 }
314325 return null
315326}
You can’t perform that action at this time.
0 commit comments