Support Chinese vits models (#368)

2023-10-18 10:19:10 +08:00
parent 9efe69720d
commit 1ee79e3ff5
16 changed files with 326 additions and 62 deletions
--- a/sherpa-onnx/csrc/offline-tts-vits-impl.h
+++ b/sherpa-onnx/csrc/offline-tts-vits-impl.h
@@ -21,7 +21,7 @@ class OfflineTtsVitsImpl : public OfflineTtsImpl {
  explicit OfflineTtsVitsImpl(const OfflineTtsConfig &config)
      : model_(std::make_unique<OfflineTtsVitsModel>(config.model)),
        lexicon_(config.model.vits.lexicon, config.model.vits.tokens,
-                 model_->Punctuations()) {}
+                 model_->Punctuations(), model_->Language()) {}

  GeneratedAudio Generate(const std::string &text,
                          int64_t sid = 0) const override {