csukuangfj commited on
Commit
bd7955f
1 Parent(s): 2978c02

add more english models

Browse files
Files changed (1) hide show
  1. model.py +21 -5
model.py CHANGED
@@ -166,14 +166,14 @@ def _get_vits_piper_de_DE(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
166
 
167
 
168
  @lru_cache(maxsize=10)
169
- def _get_vits_piper_en_US_lessac_medium(
170
- repo_id: str, speed: float
171
- ) -> sherpa_onnx.OfflineTts:
172
  assert repo_id == "csukuangfj/vits-piper-en_US-lessac-medium"
 
 
173
 
174
  model = get_file(
175
  repo_id=repo_id,
176
- filename="en_US-lessac-medium.onnx",
177
  subfolder=".",
178
  )
179
 
@@ -265,7 +265,23 @@ chinese_models = {
265
 
266
  english_models = {
267
  "csukuangfj/vits-vctk": _get_vits_vctk,
268
- "csukuangfj/vits-piper-en_US-lessac-medium": _get_vits_piper_en_US_lessac_medium,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
269
  "csukuangfj/vits-ljs": _get_vits_ljs,
270
  }
271
 
 
166
 
167
 
168
  @lru_cache(maxsize=10)
169
+ def _get_vits_piper_en_US(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
 
 
170
  assert repo_id == "csukuangfj/vits-piper-en_US-lessac-medium"
171
+ n = len("vits-piper-")
172
+ name = repo_id.split("/")[1][n:]
173
 
174
  model = get_file(
175
  repo_id=repo_id,
176
+ filename=f"{name}.onnx",
177
  subfolder=".",
178
  )
179
 
 
265
 
266
  english_models = {
267
  "csukuangfj/vits-vctk": _get_vits_vctk,
268
+ "csukuangfj/vits-piper-en_US-amy-low": _get_vits_piper_en_US,
269
+ "csukuangfj/vits-piper-en_US-amy-medium": _get_vits_piper_en_US,
270
+ "csukuangfj/vits-piper-en_US-arctic-medium": _get_vits_piper_en_US,
271
+ "csukuangfj/vits-piper-en_US-danny-low": _get_vits_piper_en_US,
272
+ "csukuangfj/vits-piper-en_US-hfc_male-medium": _get_vits_piper_en_US,
273
+ "csukuangfj/vits-piper-en_US-joe-medium": _get_vits_piper_en_US,
274
+ "csukuangfj/vits-piper-en_US-kathleen-low": _get_vits_piper_en_US,
275
+ "csukuangfj/vits-piper-en_US-kusal-medium": _get_vits_piper_en_US,
276
+ "csukuangfj/vits-piper-en_US-l2arctic-medium": _get_vits_piper_en_US,
277
+ "csukuangfj/vits-piper-en_US-lessac-low": _get_vits_piper_en_US,
278
+ "csukuangfj/vits-piper-en_US-lessac-medium": _get_vits_piper_en_US,
279
+ "csukuangfj/vits-piper-en_US-lessac-high": _get_vits_piper_en_US,
280
+ "csukuangfj/vits-piper-en_US-libritts-high": _get_vits_piper_en_US,
281
+ "csukuangfj/vits-piper-en_US-libritts_r-medium": _get_vits_piper_en_US,
282
+ "csukuangfj/vits-piper-en_US-ryan-low": _get_vits_piper_en_US,
283
+ "csukuangfj/vits-piper-en_US-ryan-medium": _get_vits_piper_en_US,
284
+ "csukuangfj/vits-piper-en_US-ryan-high": _get_vits_piper_en_US,
285
  "csukuangfj/vits-ljs": _get_vits_ljs,
286
  }
287