Skip to content

Commit 2e9f9dd

Browse files
authored
Add non-streaming zipformer Android APK (k2-fsa#1052)
1 parent f652147 commit 2e9f9dd

File tree

2 files changed

+34
-0
lines changed

2 files changed

+34
-0
lines changed

scripts/apk/generate-vad-asr-apk-script.py

+21
Original file line numberDiff line numberDiff line change
@@ -214,6 +214,27 @@ def get_models():
214214
215215
ls -lh
216216
217+
popd
218+
""",
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+
217238
popd
218239
""",
219240
),

sherpa-onnx/kotlin-api/OfflineRecognizer.kt

+13
Original file line numberDiff line numberDiff line change
@@ -297,6 +297,19 @@ fun getOfflineModelConfig(type: Int): OfflineModelConfig? {
297297
modelType = "zipformer2",
298298
)
299299
}
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+
}
300313
}
301314
return null
302315
}

0 commit comments

Comments
 (0)