Add Kotlin and Java API for FireRedAsr AED model (#1870)

This commit is contained in:
Fangjun Kuang
2025-02-17 10:50:25 +08:00
committed by GitHub
parent 316424b382
commit d148860d2c
10 changed files with 188 additions and 1 deletions

View File

@@ -33,6 +33,11 @@ data class OfflineWhisperModelConfig(
var tailPaddings: Int = 1000, // Padding added at the end of the samples
)
data class OfflineFireRedAsrModelConfig(
var encoder: String = "",
var decoder: String = "",
)
data class OfflineMoonshineModelConfig(
var preprocessor: String = "",
var encoder: String = "",
@@ -50,6 +55,7 @@ data class OfflineModelConfig(
var transducer: OfflineTransducerModelConfig = OfflineTransducerModelConfig(),
var paraformer: OfflineParaformerModelConfig = OfflineParaformerModelConfig(),
var whisper: OfflineWhisperModelConfig = OfflineWhisperModelConfig(),
var fireRedAsr: OfflineFireRedAsrModelConfig = OfflineFireRedAsrModelConfig(),
var moonshine: OfflineMoonshineModelConfig = OfflineMoonshineModelConfig(),
var nemo: OfflineNemoEncDecCtcModelConfig = OfflineNemoEncDecCtcModelConfig(),
var senseVoice: OfflineSenseVoiceModelConfig = OfflineSenseVoiceModelConfig(),
@@ -464,6 +470,17 @@ fun getOfflineModelConfig(type: Int): OfflineModelConfig? {
modelType = "transducer",
)
}
24 -> {
val modelDir = "sherpa-onnx-fire-red-asr-large-zh_en-2025-02-16"
return OfflineModelConfig(
fireRedAsr = OfflineFireRedAsrModelConfig(
encoder = "$modelDir/encoder.int8.onnx",
decoder = "$modelDir/decoder.int8.onnx",
),
tokens = "$modelDir/tokens.txt",
)
}
}
return null
}