Support VITS models from icefall. (#625)

2024-03-01 19:48:38 +08:00
parent 93836ff451
commit d56964371c
5 changed files with 13 additions and 4 deletions
--- a/sherpa-onnx/csrc/offline-tts-vits-model.cc
+++ b/sherpa-onnx/csrc/offline-tts-vits-model.cc
@@ -110,6 +110,10 @@ class OfflineTtsVitsModel::Impl {
    if (comment.find("coqui") != std::string::npos) {
      meta_data_.is_coqui = true;
    }
+
+    if (comment.find("icefall") != std::string::npos) {
+      meta_data_.is_icefall = true;
+    }
  }

  Ort::Value RunVitsPiperOrCoqui(Ort::Value x, int64_t sid, float speed) {