csukuangfj commited on
Commit
de8dd8a
·
1 Parent(s): edff7b9

add kitten v0.2

Browse files
Files changed (1) hide show
  1. model.py +5 -1
model.py CHANGED
@@ -123,7 +123,10 @@ def _get_vits_ljs(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
123
  def _get_kitten(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
124
  data_dir = "/tmp/espeak-ng-data"
125
  repo_id = repo_id.split("|")[0]
126
- assert repo_id in ("csukuangfj/kitten-nano-en-v0_1-fp16",), repo_id
 
 
 
127
 
128
  model = get_file(
129
  repo_id=repo_id,
@@ -740,6 +743,7 @@ chinese_models = {
740
 
741
  english_models = {
742
  "csukuangfj/kokoro-en-v0_19|11 speakers": _get_kokoro,
 
743
  "csukuangfj/kitten-nano-en-v0_1-fp16|8 speakers": _get_kitten,
744
  "csukuangfj/vits-piper-en_US-glados-high|1 speaker": _get_vits_piper,
745
  "csukuangfj/vits-piper-en_US-glados|1 speaker": _get_vits_piper,
 
123
  def _get_kitten(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
124
  data_dir = "/tmp/espeak-ng-data"
125
  repo_id = repo_id.split("|")[0]
126
+ assert repo_id in (
127
+ "csukuangfj/kitten-nano-en-v0_1-fp16",
128
+ "csukuangfj/kitten-nano-en-v0_2-fp16",
129
+ ), repo_id
130
 
131
  model = get_file(
132
  repo_id=repo_id,
 
743
 
744
  english_models = {
745
  "csukuangfj/kokoro-en-v0_19|11 speakers": _get_kokoro,
746
+ "csukuangfj/kitten-nano-en-v0_2-fp16|8 speakers": _get_kitten,
747
  "csukuangfj/kitten-nano-en-v0_1-fp16|8 speakers": _get_kitten,
748
  "csukuangfj/vits-piper-en_US-glados-high|1 speaker": _get_vits_piper,
749
  "csukuangfj/vits-piper-en_US-glados|1 speaker": _get_vits_piper,