Skip to content

Commit

Permalink
Build Android APKs for VITS models from Coqui-ai/TTS (k2-fsa#419)
Browse files Browse the repository at this point in the history
Support building Android APKs for VITS models from
https://github.com/coqui-ai/TTS

Please see
https://k2-fsa.github.io/sherpa/onnx/tts/apk.html
  • Loading branch information
csukuangfj authored Nov 11, 2023
1 parent c9f8c4f commit bcfc7ad
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions scripts/apk/generate-tts-apk-script.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ class TtsModel:
model_dir: str
model_name: str
lang: str # en, zh, fr, de, etc.
rule_fsts: Optional[List[str]] = (None,)
rule_fsts: Optional[List[str]] = None


def get_all_models() -> List[TtsModel]:
Expand Down Expand Up @@ -89,8 +89,13 @@ def get_all_models() -> List[TtsModel]:
lang="zh",
rule_fsts="vits-zh-hf-theresa/rule.fst",
),
# English (US)
# English (coqui-ai/TTS)
# fmt: off
TtsModel(model_dir="vits-coqui-en-ljspeech", model_name="model.onnx", lang="en"),
TtsModel(model_dir="vits-coqui-en-ljspeech-neon", model_name="model.onnx", lang="en"),
TtsModel(model_dir="vits-coqui-en-vctk", model_name="model.onnx", lang="en"),
# TtsModel(model_dir="vits-coqui-en-jenny", model_name="model.onnx", lang="en"),
# English (US)
TtsModel(model_dir="vits-vctk", model_name="vits-vctk.onnx", lang="en"),
TtsModel(model_dir="vits-ljs", model_name="vits-ljs.onnx", lang="en"),
TtsModel(model_dir="vits-piper-en_US-amy-low", model_name="en_US-amy-low.onnx", lang="en",),
Expand Down

0 comments on commit bcfc7ad

Please sign in to comment.