Add Go API for FireRedAsr AED Model (#1879)

This commit is contained in:
Fangjun Kuang
2025-02-17 16:04:07 +08:00
committed by GitHub
parent b5d89d7bcb
commit 87a968b55d
5 changed files with 40 additions and 0 deletions

View File

@@ -0,0 +1 @@
../../../../go-api-examples/non-streaming-decode-files/run-fire-red-asr.sh

View File

@@ -385,6 +385,11 @@ type OfflineWhisperModelConfig struct {
TailPaddings int
}
type OfflineFireRedAsrModelConfig struct {
Encoder string
Decoder string
}
type OfflineMoonshineModelConfig struct {
Preprocessor string
Encoder string
@@ -416,6 +421,7 @@ type OfflineModelConfig struct {
Tdnn OfflineTdnnModelConfig
SenseVoice OfflineSenseVoiceModelConfig
Moonshine OfflineMoonshineModelConfig
FireRedAsr OfflineFireRedAsrModelConfig
Tokens string // Path to tokens.txt
// Number of threads to use for neural network computation
@@ -538,6 +544,12 @@ func NewOfflineRecognizer(config *OfflineRecognizerConfig) *OfflineRecognizer {
c.model_config.moonshine.cached_decoder = C.CString(config.ModelConfig.Moonshine.CachedDecoder)
defer C.free(unsafe.Pointer(c.model_config.moonshine.cached_decoder))
c.model_config.fire_red_asr.encoder = C.CString(config.ModelConfig.FireRedAsr.Encoder)
defer C.free(unsafe.Pointer(c.model_config.fire_red_asr.encoder))
c.model_config.fire_red_asr.decoder = C.CString(config.ModelConfig.FireRedAsr.Decoder)
defer C.free(unsafe.Pointer(c.model_config.fire_red_asr.decoder))
c.model_config.tokens = C.CString(config.ModelConfig.Tokens)
defer C.free(unsafe.Pointer(c.model_config.tokens))