EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Validate input sid (#369)

Browse Source
This commit is contained in:
Fangjun Kuang
2023-10-18 14:02:01 +08:00
committed by GitHub
parent 1ee79e3ff5
commit 8545c3b7f0
8 changed files with 51 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
sherpa-onnx/csrc/offline-tts-vits-model.cc
View File

@@ -85,6 +85,7 @@ class OfflineTtsVitsModel::Impl {
std::string Punctuations() const { return punctuations_; }
std::string Language() const { return language_; }
int32_t NumSpeakers() const { return num_speakers_; }
private:
void Init(void *model_data, size_t model_data_length) {
@@ -107,7 +108,7 @@ class OfflineTtsVitsModel::Impl {
Ort::AllocatorWithDefaultOptions allocator; // used in the macro below
SHERPA_ONNX_READ_META_DATA(sample_rate_, "sample_rate");
SHERPA_ONNX_READ_META_DATA(add_blank_, "add_blank");
SHERPA_ONNX_READ_META_DATA(n_speakers_, "n_speakers");
SHERPA_ONNX_READ_META_DATA(num_speakers_, "n_speakers");
SHERPA_ONNX_READ_META_DATA_STR(punctuations_, "punctuation");
SHERPA_ONNX_READ_META_DATA_STR(language_, "language");
}
@@ -128,7 +129,7 @@ class OfflineTtsVitsModel::Impl {
int32_t sample_rate_;
int32_t add_blank_;
int32_t n_speakers_;
int32_t num_speakers_;
std::string punctuations_;
std::string language_;
};
@@ -152,4 +153,8 @@ std::string OfflineTtsVitsModel::Punctuations() const {
std::string OfflineTtsVitsModel::Language() const { return impl_->Language(); }
int32_t OfflineTtsVitsModel::NumSpeakers() const {
return impl_->NumSpeakers();
}
} // namespace sherpa_onnx