正在显示
2 个修改的文件
包含
34 行增加
和
0 行删除
| @@ -90,6 +90,29 @@ def get_models(): | @@ -90,6 +90,29 @@ def get_models(): | ||
| 90 | """, | 90 | """, |
| 91 | ), | 91 | ), |
| 92 | Model( | 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 | + ), | ||
| 115 | + Model( | ||
| 93 | model_name="icefall-asr-zipformer-wenetspeech-20230615", | 116 | model_name="icefall-asr-zipformer-wenetspeech-20230615", |
| 94 | idx=4, | 117 | idx=4, |
| 95 | lang="zh", | 118 | lang="zh", |
| @@ -310,6 +310,17 @@ fun getOfflineModelConfig(type: Int): OfflineModelConfig? { | @@ -310,6 +310,17 @@ fun getOfflineModelConfig(type: Int): OfflineModelConfig? { | ||
| 310 | modelType = "transducer", | 310 | modelType = "transducer", |
| 311 | ) | 311 | ) |
| 312 | } | 312 | } |
| 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 | + } | ||
| 313 | } | 324 | } |
| 314 | return null | 325 | return null |
| 315 | } | 326 | } |
-
请 注册 或 登录 后发表评论