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

Before consolidated main

Browse Source
This commit is contained in:
Michael Hansen
2023-01-09 16:35:32 -06:00
parent 6c949aa5c5
commit dfa03d80e6
3 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/cpp/main.cpp
View File

@@ -188,7 +188,7 @@ void parseArgs(int argc, char *argv[], RunConfig &runConfig) {
runConfig.outputPath = filesystem::path(argv[++i]);
} else if (arg == "-s" || arg == "--speaker") {
ensureArg(argc, argv, i);
runConfig.speakerId = (larynx::SpeakerId)stoi(argv[++i]);
runConfig.speakerId = (larynx::SpeakerId)stoll(argv[++i]);
} else if (arg == "-h" || arg == "--help") {
printUsage(argv);
exit(0);

9
src/cpp/model.hpp
View File

@@ -16,15 +16,16 @@ struct ModelSession {
vector<char *> outputNames;
Ort::AllocatorWithDefaultOptions allocator;
Ort::SessionOptions options;
Ort::Env env;
ModelSession() : onnx(nullptr){};
};
void loadModel(string modelPath, ModelSession &session) {
Ort::Env env(OrtLoggingLevel::ORT_LOGGING_LEVEL_WARNING,
instanceName.c_str());
env.DisableTelemetryEvents();
session.env = Ort::Env(OrtLoggingLevel::ORT_LOGGING_LEVEL_WARNING,
instanceName.c_str());
session.env.DisableTelemetryEvents();
// Slows down performance by ~2x
// session.options.SetIntraOpNumThreads(1);
@@ -44,7 +45,7 @@ void loadModel(string modelPath, ModelSession &session) {
session.options.DisableProfiling();
auto startTime = chrono::steady_clock::now();
session.onnx = Ort::Session(env, modelPath.c_str(), session.options);
session.onnx = Ort::Session(session.env, modelPath.c_str(), session.options);
auto endTime = chrono::steady_clock::now();
auto loadDuration = chrono::duration<double>(endTime - startTime);

2
src/cpp/synthesize.hpp
View File

@@ -56,7 +56,7 @@ void synthesize(SynthesisConfig &synthesisConfig, ModelSession &session,
if (synthesisConfig.speakerId) {
// Add speaker id
vector<int64_t> speakerId{(int64_t)synthesisConfig.speakerId.value()};
vector<int64_t> speakerIdShape{1};
vector<int64_t> speakerIdShape{(int64_t)speakerId.size()};
inputTensors.push_back(Ort::Value::CreateTensor<int64_t>(
memoryInfo, speakerId.data(), speakerId.size(), speakerIdShape.data(),
speakerIdShape.size()));