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/piper-phonemize-lexicon.cc
+++ b/sherpa-onnx/csrc/piper-phonemize-lexicon.cc
@@ -236,7 +236,7 @@ std::vector<std::vector<int64_t>> PiperPhonemizeLexicon::ConvertTextToTokenIds(

  std::vector<int64_t> phoneme_ids;

-  if (meta_data_.is_piper) {
+  if (meta_data_.is_piper || meta_data_.is_icefall) {
    for (const auto &p : phonemes) {
      phoneme_ids = PiperPhonemesToIds(token2id_, p);
      ans.push_back(std::move(phoneme_ids));