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

Add multispeaker

Browse Source
This commit is contained in:
Michael Hansen
2023-01-05 21:47:08 -06:00
parent 06a154a4ed
commit a7fe73390e
8 changed files with 68 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/cpp/synthesize.hpp
View File

@@ -53,6 +53,15 @@ void synthesize(SynthesisConfig &synthesisConfig, ModelSession &session,
Ort::Value::CreateTensor<float>(memoryInfo, scales.data(), scales.size(),
scalesShape.data(), scalesShape.size()));
if (synthesisConfig.speakerId) {
// Add speaker id
vector<int64_t> speakerId{(int64_t)synthesisConfig.speakerId.value()};
vector<int64_t> speakerIdShape{1};
inputTensors.push_back(Ort::Value::CreateTensor<int64_t>(
memoryInfo, speakerId.data(), speakerId.size(), speakerIdShape.data(),
speakerIdShape.size()));
}
// Infer
auto startTime = chrono::steady_clock::now();
auto outputTensors =