Add jieba for Chinese TTS models (#797)

This commit is contained in:
Fangjun Kuang
2024-04-21 14:47:13 +08:00
committed by GitHub
parent 2e0ee0e8c8
commit 6b353bfb42
14 changed files with 513 additions and 8 deletions

View File

@@ -20,6 +20,9 @@ struct OfflineTtsVitsModelConfig {
// data_dir is for piper-phonemize, which uses espeak-ng
std::string data_dir;
// Used for Chinese TTS models using jieba
std::string dict_dir;
float noise_scale = 0.667;
float noise_scale_w = 0.8;
float length_scale = 1;
@@ -33,12 +36,14 @@ struct OfflineTtsVitsModelConfig {
const std::string &lexicon,
const std::string &tokens,
const std::string &data_dir,
const std::string &dict_dir,
float noise_scale = 0.667,
float noise_scale_w = 0.8, float length_scale = 1)
: model(model),
lexicon(lexicon),
tokens(tokens),
data_dir(data_dir),
dict_dir(dict_dir),
noise_scale(noise_scale),
noise_scale_w(noise_scale_w),
length_scale(length_scale) {}