Fix punctuations in tts (k2-fsa#417)

XiaYucca · Nov 10, 2023 · c9f8c4f · c9f8c4f
1 parent d9e25ac
commit c9f8c4f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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;