csukuangfj commited on
Commit
23e7dca
1 Parent(s): 5e1c24b

small fixes

Browse files
Files changed (2) hide show
  1. model.py +8 -1
  2. requirements.txt +2 -2
model.py CHANGED
@@ -176,6 +176,12 @@ def _get_vits_zh_aishell3(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
176
  subfolder=".",
177
  )
178
 
 
 
 
 
 
 
179
  tts_config = sherpa_onnx.OfflineTtsConfig(
180
  model=sherpa_onnx.OfflineTtsModelConfig(
181
  vits=sherpa_onnx.OfflineTtsVitsModelConfig(
@@ -187,7 +193,8 @@ def _get_vits_zh_aishell3(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
187
  provider="cpu",
188
  debug=True,
189
  num_threads=2,
190
- )
 
191
  )
192
  tts = sherpa_onnx.OfflineTts(tts_config)
193
 
 
176
  subfolder=".",
177
  )
178
 
179
+ rule_fst = get_file(
180
+ repo_id=repo_id,
181
+ filename="rule.fst",
182
+ subfolder=".",
183
+ )
184
+
185
  tts_config = sherpa_onnx.OfflineTtsConfig(
186
  model=sherpa_onnx.OfflineTtsModelConfig(
187
  vits=sherpa_onnx.OfflineTtsVitsModelConfig(
 
193
  provider="cpu",
194
  debug=True,
195
  num_threads=2,
196
+ ),
197
+ rule_fsts=rule_fst,
198
  )
199
  tts = sherpa_onnx.OfflineTts(tts_config)
200
 
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- # https://huggingface.co/csukuangfj/wheels/resolve/main/sherpa_onnx-1.8.6-cp38-cp38-linux_x86_64.whl
2
- sherpa-onnx
3
 
4
  soundfile
 
1
+ https://huggingface.co/csukuangfj/wheels/resolve/main/sherpa_onnx-1.8.7-cp38-cp38-linux_x86_64.whl
2
+ #sherpa-onnx
3
 
4
  soundfile