Add Kotlin and Java API for FireRedAsr AED model (#1870)
This commit is contained in:
@@ -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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user