support whisper language (#678)

This commit is contained in:
Viggo
2024-03-20 10:16:22 +08:00
committed by GitHub
parent 6571fc9552
commit 842d04d7ae
2 changed files with 4 additions and 0 deletions

View File

@@ -298,6 +298,9 @@ SherpaOnnxOfflineRecognizer *CreateOfflineRecognizer(
recognizer_config.model_config.whisper.decoder = recognizer_config.model_config.whisper.decoder =
SHERPA_ONNX_OR(config->model_config.whisper.decoder, ""); SHERPA_ONNX_OR(config->model_config.whisper.decoder, "");
recognizer_config.model_config.whisper.language =
SHERPA_ONNX_OR(config->model_config.whisper.language, "");
recognizer_config.model_config.tdnn.model = recognizer_config.model_config.tdnn.model =
SHERPA_ONNX_OR(config->model_config.tdnn.model, ""); SHERPA_ONNX_OR(config->model_config.tdnn.model, "");

View File

@@ -332,6 +332,7 @@ SHERPA_ONNX_API typedef struct SherpaOnnxOfflineNemoEncDecCtcModelConfig {
SHERPA_ONNX_API typedef struct SherpaOnnxOfflineWhisperModelConfig { SHERPA_ONNX_API typedef struct SherpaOnnxOfflineWhisperModelConfig {
const char *encoder; const char *encoder;
const char *decoder; const char *decoder;
const char *language;
} SherpaOnnxOfflineWhisperModelConfig; } SherpaOnnxOfflineWhisperModelConfig;
SHERPA_ONNX_API typedef struct SherpaOnnxOfflineTdnnModelConfig { SHERPA_ONNX_API typedef struct SherpaOnnxOfflineTdnnModelConfig {