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

Upload more onnx ASR models (#2141)

Browse Source
This commit is contained in:
Fangjun Kuang
2025-04-21 18:57:41 +08:00
committed by GitHub
parent 6cabaa11bf
commit 7cbb1bc433
3 changed files with 338 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
sherpa-onnx/kotlin-api/OfflineRecognizer.kt
View File

@@ -496,6 +496,55 @@ fun getOfflineModelConfig(type: Int): OfflineModelConfig? {
tokens = "$modelDir/tokens.txt",
)
}
26 -> {
val modelDir = "sherpa-onnx-zipformer-vi-int8-2025-04-20"
return OfflineModelConfig(
transducer = OfflineTransducerModelConfig(
encoder = "$modelDir/encoder-epoch-12-avg-8.int8.onnx",
decoder = "$modelDir/decoder-epoch-12-avg-8.onnx",
joiner = "$modelDir/joiner-epoch-12-avg-8.int8.onnx",
),
tokens = "$modelDir/tokens.txt",
modelType = "transducer",
)
}
27 -> {
val modelDir = "sherpa-onnx-nemo-ctc-giga-am-v2-russian-2025-04-19"
return OfflineModelConfig(
nemo = OfflineNemoEncDecCtcModelConfig(
model = "$modelDir/model.int8.onnx",
),
tokens = "$modelDir/tokens.txt",
)
}
28 -> {
val modelDir = "sherpa-onnx-nemo-transducer-giga-am-v2-russian-2025-04-19"
return OfflineModelConfig(
transducer = OfflineTransducerModelConfig(
encoder = "$modelDir/encoder.int8.onnx",
decoder = "$modelDir/decoder.onnx",
joiner = "$modelDir/joiner.onnx",
),
tokens = "$modelDir/tokens.txt",
modelType = "nemo_transducer",
)
}
29 -> {
val modelDir = "sherpa-onnx-zipformer-ru-int8-2025-04-20"
return OfflineModelConfig(
transducer = OfflineTransducerModelConfig(
encoder = "$modelDir/encoder.int8.onnx",
decoder = "$modelDir/decoder.onnx",
joiner = "$modelDir/joiner.int8.onnx",
),
tokens = "$modelDir/tokens.txt",
modelType = "transducer",
)
}
}
return null
}