Fix punctuations in tts (#417)

k2-fsa · Nov 10, 2023 · 47947ff · 47947ff
1 parent 61341b7
commit 47947ff
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;