Support VITS TTS models from coqui-ai/TTS (#416)

* Support VITS TTS models from coqui-ai/TTS * release v1.8.9
2023-11-10 16:24:11 +08:00
parent ab0e830bee
commit 61341b7187
3 changed files with 25 additions and 9 deletions
--- a/sherpa-onnx/csrc/offline-tts-vits-model.cc
+++ b/sherpa-onnx/csrc/offline-tts-vits-model.cc
@@ -81,7 +81,8 @@ class OfflineTtsVitsModel::Impl {

    std::string comment;
    SHERPA_ONNX_READ_META_DATA_STR(comment, "comment");
-    if (comment.find("piper") != std::string::npos) {
+    if (comment.find("piper") != std::string::npos ||
+        comment.find("coqui") != std::string::npos) {
      is_piper_ = true;
    }
  }