Add CXX API for FireRedAsr (#1872)

This commit is contained in:
Fangjun Kuang
2025-02-17 11:46:13 +08:00
committed by GitHub
parent 193d31333c
commit 1d49dd2fb0
5 changed files with 129 additions and 5 deletions

View File

@@ -241,6 +241,11 @@ OfflineRecognizer OfflineRecognizer::Create(
c.model_config.moonshine.cached_decoder =
config.model_config.moonshine.cached_decoder.c_str();
c.model_config.fire_red_asr.encoder =
config.model_config.fire_red_asr.encoder.c_str();
c.model_config.fire_red_asr.decoder =
config.model_config.fire_red_asr.decoder.c_str();
c.lm_config.model = config.lm_config.model.c_str();
c.lm_config.scale = config.lm_config.scale;

View File

@@ -214,6 +214,11 @@ struct SHERPA_ONNX_API OfflineWhisperModelConfig {
int32_t tail_paddings = -1;
};
struct SHERPA_ONNX_API OfflineFireRedAsrModelConfig {
std::string encoder;
std::string decoder;
};
struct SHERPA_ONNX_API OfflineTdnnModelConfig {
std::string model;
};
@@ -248,6 +253,7 @@ struct SHERPA_ONNX_API OfflineModelConfig {
std::string telespeech_ctc;
OfflineSenseVoiceModelConfig sense_voice;
OfflineMoonshineModelConfig moonshine;
OfflineFireRedAsrModelConfig fire_red_asr;
};
struct SHERPA_ONNX_API OfflineLMConfig {