add one more model

Author: csukuangfj

.github/workflows/export-piper.yaml

@@ -20,15 +20,15 @@ jobs:
       matrix:
         os: [ubuntu-latest]
         python-version: ["3.10"]
-        # total: ["20"]
-        # index: [
-        #   "0", "1", "2", "3", "4", "5", "6", "7", "8", "9",
-        #   "10", "11", "12", "13", "14", "15", "16", "17", "18", "19",
-        #    ]
+        total: ["20"]
+        index: [
+          "0", "1", "2", "3", "4", "5", "6", "7", "8", "9",
+          "10", "11", "12", "13", "14", "15", "16", "17", "18", "19",
+           ]
         # total: ["2"]
         # index: ["0", "1"]
-        total: ["1"]
-        index: ["0"]
+        # total: ["1"]
+        # index: ["0"]
         # total: ["5"]
         # index: ["0", "1", "2", "3", "4"]
 
@@ -185,6 +185,9 @@ jobs:
               vits-piper-zh_CN-xiao_ya-medium
               vits-piper-zh_CN-xiao_ya-medium-int8
               vits-piper-zh_CN-xiao_ya-medium-fp16
+              vits-piper-zh_CN-chaowen-medium
+              vits-piper-zh_CN-chaowen-medium-int8
+              vits-piper-zh_CN-chaowen-medium-fp16
             )
             for d in ${dirs[@]}; do
               src=scripts/piper/release/$d

scripts/apk/generate-tts-apk-script.py

@@ -278,10 +278,13 @@ def get_piper_models() -> List[TtsModel]:
         TtsModel(model_dir="vits-piper-vi_VN-vivos-x_low"),
         TtsModel(model_dir="vits-piper-zh_CN-huayan-medium"),
         TtsModel(model_dir="vits-piper-zh_CN-xiao_ya-medium"),
+        TtsModel(model_dir="vits-piper-zh_CN-chaowen-medium"),
     ]
 
     for m in models:
-        if "zh_CN-xiao_ya" in m.model_dir:
+        if "zh_CN" in m.model_dir and (
+            "xiao_ya" in m.model_dir or "chaowen" in m.model_dir
+        ):
             m.lexion = f"{m.model_dir}/lexicon.txt"
             m.rule_fsts = f"{m.model_dir}/phone.fst,{m.model_dir}/date.fst,{m.model_dir}/number.fst"
         else:

scripts/piper/generate.py

@@ -1651,6 +1651,7 @@ def get_vi_models():
 def get_zh_models():
     zh_CN = [
         PiperModel(name="xiao_ya", kind="medium", sr=22050, ns=1),
+        PiperModel(name="chaowen", kind="medium", sr=22050, ns=1),
     ]
 
     for m in zh_CN:
@@ -2144,8 +2145,7 @@ def get_all_models():
     ans += get_tr_models()
     ans += get_uk_models()
     ans += get_vi_models()
-
-    ans = get_zh_models()
+    ans += get_zh_models()
 
     for i, m in enumerate(ans):
         m.index = i
@@ -2219,7 +2219,7 @@ def main():
                 "rule_fsts": "",
             }
 
-            if m.lang == "zh_CN" and m.name == "xiao_ya":
+            if m.lang == "zh_CN" and m.name in ("xiao_ya", "chaowen"):
                 d = {
                     "model": f"{model_dir}/{m.model_name}",
                     "data_dir": "",

scripts/piper/generate.sh.in

@@ -57,7 +57,7 @@ mv -v README $dst/ || true
 mv -v README.md $dst/ || true
 mv -v LICENSE.txt $dst/ || true
 mv -v *.json  $dst/
-if [[ $model_name != *zh_CN-xiao_ya-medium* ]]; then
+if [[ ! $model_name =~ zh_CN-(xiao_ya|chaowen)-medium ]]; then
   cp -a ./espeak-ng-data $dst/
 else
   cp lexicon-zh-g2pw.txt $dst/lexicon.txt