Committed by
GitHub
Add non-streaming zipformer Android APK (#1052)
正在显示
2 个修改的文件
包含
34 行增加
和
0 行删除
| @@ -217,6 +217,27 @@ def get_models(): | @@ -217,6 +217,27 @@ def get_models(): | ||
| 217 | popd | 217 | popd |
| 218 | """, | 218 | """, |
| 219 | ), | 219 | ), |
| 220 | + Model( | ||
| 221 | + model_name="sherpa-onnx-zipformer-korean-2024-06-24", | ||
| 222 | + idx=13, | ||
| 223 | + lang="ko", | ||
| 224 | + short_name="zipformer", | ||
| 225 | + cmd=""" | ||
| 226 | + pushd $model_name | ||
| 227 | + | ||
| 228 | + rm -rfv test_wavs | ||
| 229 | + rm -fv README.md | ||
| 230 | + rm -fv bpe.model | ||
| 231 | + | ||
| 232 | + rm encoder-epoch-99-avg-1.onnx | ||
| 233 | + rm decoder-epoch-99-avg-1.int8.onnx | ||
| 234 | + rm joiner-epoch-99-avg-1.onnx | ||
| 235 | + | ||
| 236 | + ls -lh | ||
| 237 | + | ||
| 238 | + popd | ||
| 239 | + """, | ||
| 240 | + ), | ||
| 220 | ] | 241 | ] |
| 221 | return models | 242 | return models |
| 222 | 243 |
| @@ -297,6 +297,19 @@ fun getOfflineModelConfig(type: Int): OfflineModelConfig? { | @@ -297,6 +297,19 @@ fun getOfflineModelConfig(type: Int): OfflineModelConfig? { | ||
| 297 | modelType = "zipformer2", | 297 | modelType = "zipformer2", |
| 298 | ) | 298 | ) |
| 299 | } | 299 | } |
| 300 | + | ||
| 301 | + 13 -> { | ||
| 302 | + val modelDir = "sherpa-onnx-zipformer-korean-2024-06-24" | ||
| 303 | + return OfflineModelConfig( | ||
| 304 | + transducer = OfflineTransducerModelConfig( | ||
| 305 | + encoder = "$modelDir/encoder-epoch-99-avg-1.int8.onnx", | ||
| 306 | + decoder = "$modelDir/decoder-epoch-99-avg-1.onnx", | ||
| 307 | + joiner = "$modelDir/joiner-epoch-99-avg-1.int8.onnx", | ||
| 308 | + ), | ||
| 309 | + tokens = "$modelDir/tokens.txt", | ||
| 310 | + modelType = "zipformer2", | ||
| 311 | + ) | ||
| 312 | + } | ||
| 300 | } | 313 | } |
| 301 | return null | 314 | return null |
| 302 | } | 315 | } |
-
请 注册 或 登录 后发表评论