Fix punctuations in tts (#417)

2023-11-10 17:09:48 +08:00
parent 61341b7187
commit 47947ffae9
1 changed files with 3 additions and 1 deletions
--- a/sherpa-onnx/csrc/lexicon.cc
+++ b/sherpa-onnx/csrc/lexicon.cc
@@ -214,7 +214,9 @@ std::vector<int64_t> Lexicon::ConvertTextToTokenIdsChinese(

  for (const auto &w : words) {
    if (punctuations_.count(w)) {
-      if (sil != -1) {
+      if (token2id_.count(w)) {
+        ans.push_back(token2id_.at(w));
+      } else if (sil != -1) {
        ans.push_back(sil);
      }
      continue;