auracaster/piper
3
0
Fork 0
mirror of https://github.com/pstrueb/piper.git synced 2026-04-29 19:24:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add "speaker" to JSON input

Browse Source
This commit is contained in:
Michael Hansen
2023-07-10 12:46:46 -05:00
parent 0a5d144656
commit 42d14ef21f
4 changed files with 27 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/cpp/piper.cpp
View File

@@ -163,6 +163,19 @@ void parseModelConfig(json &configRoot, ModelConfig &modelConfig) {
modelConfig.numSpeakers = configRoot["num_speakers"].get<SpeakerId>();
if (configRoot.contains("speaker_id_map")) {
if (!modelConfig.speakerIdMap) {
modelConfig.speakerIdMap.emplace();
}
auto speakerIdMapValue = configRoot["speaker_id_map"];
for (auto &speakerItem : speakerIdMapValue.items()) {
std::string speakerName = speakerItem.key();
(*modelConfig.speakerIdMap)[speakerName] =
speakerItem.value().get<SpeakerId>();
}
}
} /* parseModelConfig */
void initialize(PiperConfig &config) {