EngineX-Iluvatar/enginex_bi_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Support running sherpa-onnx with RK NPU on Android (#2124)

Browse Source
This commit is contained in:
Fangjun Kuang
2025-04-15 16:42:28 +08:00
committed by GitHub
parent 4ed8ec367e
commit e3bce847c0
10 changed files with 427 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
sherpa-onnx/kotlin-api/OnlineRecognizer.kt
View File

@@ -394,6 +394,34 @@ fun getModelConfig(type: Int): OnlineModelConfig? {
tokens = "$modelDir/tokens.txt",
)
}
1000 -> {
val modelDir = "sherpa-onnx-rk3588-streaming-zipformer-bilingual-zh-en-2023-02-20"
return OnlineModelConfig(
transducer = OnlineTransducerModelConfig(
encoder = "$modelDir/encoder.rknn",
decoder = "$modelDir/decoder.rknn",
joiner = "$modelDir/joiner.rknn",
),
tokens = "$modelDir/tokens.txt",
modelType = "zipformer",
provider = "rknn",
)
}
1001 -> {
val modelDir = "sherpa-onnx-rk3588-streaming-zipformer-small-bilingual-zh-en-2023-02-16"
return OnlineModelConfig(
transducer = OnlineTransducerModelConfig(
encoder = "$modelDir/encoder.rknn",
decoder = "$modelDir/decoder.rknn",
joiner = "$modelDir/joiner.rknn",
),
tokens = "$modelDir/tokens.txt",
modelType = "zipformer",
provider = "rknn",
)
}
}
return null
}