csukuangfj commited on
Commit
7552061
1 Parent(s): 40e123f

add 2 new models from coqui

Browse files
Files changed (1) hide show
  1. model.py +5 -0
model.py CHANGED
@@ -205,6 +205,8 @@ def _get_vits_zh_aishell3(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
205
  def _get_vits_hf(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
206
  if "fanchen" in repo_id or "vits-cantonese-hf-xiaomaiiwn" in repo_id:
207
  model = repo_id.split("/")[-1]
 
 
208
  else:
209
  model = repo_id.split("-")[-1]
210
 
@@ -293,6 +295,9 @@ chinese_models = {
293
  english_models = {
294
  "csukuangfj/vits-vctk": _get_vits_vctk, # 109 speakers
295
  "csukuangfj/vits-ljs": _get_vits_ljs,
 
 
 
296
  # piper, US
297
  "csukuangfj/vits-piper-en_US-amy-low": _get_vits_piper,
298
  "csukuangfj/vits-piper-en_US-amy-medium": _get_vits_piper,
 
205
  def _get_vits_hf(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
206
  if "fanchen" in repo_id or "vits-cantonese-hf-xiaomaiiwn" in repo_id:
207
  model = repo_id.split("/")[-1]
208
+ elif "coqui" in repo_id:
209
+ model = "model"
210
  else:
211
  model = repo_id.split("-")[-1]
212
 
 
295
  english_models = {
296
  "csukuangfj/vits-vctk": _get_vits_vctk, # 109 speakers
297
  "csukuangfj/vits-ljs": _get_vits_ljs,
298
+ # coqui-ai
299
+ "csukuangfj/vits-coqui-en-ljspeech": _get_vits_hf,
300
+ "csukuangfj/vits-coqui-en-ljspeech-neon": _get_vits_hf,
301
  # piper, US
302
  "csukuangfj/vits-piper-en_US-amy-low": _get_vits_piper,
303
  "csukuangfj/vits-piper-en_US-amy-medium": _get_vits_piper,