Support extra languages in multi-lang kokoro tts (#2303)
This commit is contained in:
@@ -1164,6 +1164,7 @@ static sherpa_onnx::OfflineTtsConfig GetOfflineTtsConfig(
|
||||
SHERPA_ONNX_OR(config->model.kokoro.dict_dir, "");
|
||||
tts_config.model.kokoro.lexicon =
|
||||
SHERPA_ONNX_OR(config->model.kokoro.lexicon, "");
|
||||
tts_config.model.kokoro.lang = SHERPA_ONNX_OR(config->model.kokoro.lang, "");
|
||||
|
||||
tts_config.model.num_threads = SHERPA_ONNX_OR(config->model.num_threads, 1);
|
||||
tts_config.model.debug = config->model.debug;
|
||||
|
||||
@@ -958,6 +958,7 @@ SHERPA_ONNX_API typedef struct SherpaOnnxOfflineTtsKokoroModelConfig {
|
||||
float length_scale; // < 1, faster in speech speed; > 1, slower in speed
|
||||
const char *dict_dir;
|
||||
const char *lexicon;
|
||||
const char *lang;
|
||||
} SherpaOnnxOfflineTtsKokoroModelConfig;
|
||||
|
||||
SHERPA_ONNX_API typedef struct SherpaOnnxOfflineTtsModelConfig {
|
||||
|
||||
@@ -366,6 +366,7 @@ OfflineTts OfflineTts::Create(const OfflineTtsConfig &config) {
|
||||
c.model.kokoro.length_scale = config.model.kokoro.length_scale;
|
||||
c.model.kokoro.dict_dir = config.model.kokoro.dict_dir.c_str();
|
||||
c.model.kokoro.lexicon = config.model.kokoro.lexicon.c_str();
|
||||
c.model.kokoro.lang = config.model.kokoro.lang.c_str();
|
||||
|
||||
c.model.num_threads = config.model.num_threads;
|
||||
c.model.debug = config.model.debug;
|
||||
|
||||
@@ -367,6 +367,7 @@ struct OfflineTtsKokoroModelConfig {
|
||||
std::string data_dir;
|
||||
std::string dict_dir;
|
||||
std::string lexicon;
|
||||
std::string lang;
|
||||
|
||||
float length_scale = 1.0; // < 1, faster in speed; > 1, slower in speed
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user