正在显示
3 个修改的文件
包含
9 行增加
和
4 行删除
| @@ -332,18 +332,23 @@ def get_vits_models() -> List[TtsModel]: | @@ -332,18 +332,23 @@ def get_vits_models() -> List[TtsModel]: | ||
| 332 | model_name="vits-zh-hf-fanchen-unity.onnx", | 332 | model_name="vits-zh-hf-fanchen-unity.onnx", |
| 333 | lang="zh", | 333 | lang="zh", |
| 334 | ), | 334 | ), |
| 335 | + TtsModel( | ||
| 336 | + model_dir="sherpa-onnx-vits-zh-ll", | ||
| 337 | + model_name="model.onnx", | ||
| 338 | + lang="zh", | ||
| 339 | + ), | ||
| 335 | ] | 340 | ] |
| 336 | 341 | ||
| 337 | rule_fsts = ["phone.fst", "date.fst", "number.fst", "new_heteronym.fst"] | 342 | rule_fsts = ["phone.fst", "date.fst", "number.fst", "new_heteronym.fst"] |
| 338 | for m in chinese_models: | 343 | for m in chinese_models: |
| 339 | s = [f"{m.model_dir}/{r}" for r in rule_fsts] | 344 | s = [f"{m.model_dir}/{r}" for r in rule_fsts] |
| 340 | - if "vits-zh-hf" in m.model_dir: | 345 | + if "vits-zh-hf" in m.model_dir or "sherpa-onnx-vits-zh-ll" == m.model_dir: |
| 341 | s = s[:-1] | 346 | s = s[:-1] |
| 342 | m.dict_dir = m.model_dir + "/dict" | 347 | m.dict_dir = m.model_dir + "/dict" |
| 343 | 348 | ||
| 344 | m.rule_fsts = ",".join(s) | 349 | m.rule_fsts = ",".join(s) |
| 345 | 350 | ||
| 346 | - if "vits-zh-hf" not in m.model_dir: | 351 | + if "vits-zh-hf" not in m.model_dir and "zh-ll" not in m.model_dir: |
| 347 | m.rule_fars = f"{m.model_dir}/rule.far" | 352 | m.rule_fars = f"{m.model_dir}/rule.far" |
| 348 | 353 | ||
| 349 | all_models = chinese_models + [ | 354 | all_models = chinese_models + [ |
-
请 注册 或 登录 后发表评论