csukuangfj commited on
Commit
35727af
·
1 Parent(s): 6259405

Add kokoro v1.1-zh

Browse files
Files changed (1) hide show
  1. model.py +6 -1
model.py CHANGED
@@ -126,6 +126,7 @@ def _get_kokoro(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
126
  assert repo_id in (
127
  "csukuangfj/kokoro-en-v0_19",
128
  "csukuangfj/kokoro-multi-lang-v1_0",
 
129
  ), repo_id
130
 
131
  model = get_file(
@@ -145,7 +146,10 @@ def _get_kokoro(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
145
  filename="voices.bin",
146
  subfolder=".",
147
  )
148
- if repo_id == "csukuangfj/kokoro-multi-lang-v1_0":
 
 
 
149
  lexicon_en = get_file(
150
  repo_id=repo_id,
151
  filename="lexicon-us-en.txt",
@@ -654,6 +658,7 @@ cantonese_models = {
654
  }
655
 
656
  chinese_english_models = {
 
657
  "csukuangfj/kokoro-multi-lang-v1_0|53 speakers": _get_kokoro,
658
  "csukuangfj/vits-melo-tts-zh_en|1": _get_vits_hf, # 1
659
  }
 
126
  assert repo_id in (
127
  "csukuangfj/kokoro-en-v0_19",
128
  "csukuangfj/kokoro-multi-lang-v1_0",
129
+ "csukuangfj/kokoro-multi-lang-v1_1",
130
  ), repo_id
131
 
132
  model = get_file(
 
146
  filename="voices.bin",
147
  subfolder=".",
148
  )
149
+ if repo_id in (
150
+ "csukuangfj/kokoro-multi-lang-v1_0",
151
+ "csukuangfj/kokoro-multi-lang-v1_1",
152
+ ):
153
  lexicon_en = get_file(
154
  repo_id=repo_id,
155
  filename="lexicon-us-en.txt",
 
658
  }
659
 
660
  chinese_english_models = {
661
+ "csukuangfj/kokoro-multi-lang-v1_1|103 speakers": _get_kokoro,
662
  "csukuangfj/kokoro-multi-lang-v1_0|53 speakers": _get_kokoro,
663
  "csukuangfj/vits-melo-tts-zh_en|1": _get_vits_hf, # 1
664
  }