Spaces:
Running
Running
csukuangfj
commited on
Commit
•
f7215c7
1
Parent(s):
d42a943
add more english models
Browse files
model.py
CHANGED
@@ -114,59 +114,7 @@ def _get_vits_ljs(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
|
114 |
|
115 |
|
116 |
@lru_cache(maxsize=10)
|
117 |
-
def
|
118 |
-
assert repo_id in (
|
119 |
-
"csukuangfj/vits-piper-de_DE-eva_k-x_low",
|
120 |
-
"csukuangfj/vits-piper-de_DE-karlsson-low",
|
121 |
-
"csukuangfj/vits-piper-de_DE-kerstin-low",
|
122 |
-
"csukuangfj/vits-piper-de_DE-pavoque-low",
|
123 |
-
"csukuangfj/vits-piper-de_DE-ramona-low",
|
124 |
-
"csukuangfj/vits-piper-de_DE-thorsten-low",
|
125 |
-
"csukuangfj/vits-piper-de_DE-thorsten-medium",
|
126 |
-
"csukuangfj/vits-piper-de_DE-thorsten-high",
|
127 |
-
"csukuangfj/vits-piper-de_DE-thorsten_emotional-medium",
|
128 |
-
)
|
129 |
-
n = len("vits-piper-")
|
130 |
-
name = repo_id.split("/")[1][n:]
|
131 |
-
|
132 |
-
model = get_file(
|
133 |
-
repo_id=repo_id,
|
134 |
-
filename=f"{name}.onnx",
|
135 |
-
subfolder=".",
|
136 |
-
)
|
137 |
-
|
138 |
-
lexicon = get_file(
|
139 |
-
repo_id=repo_id,
|
140 |
-
filename="lexicon.txt",
|
141 |
-
subfolder=".",
|
142 |
-
)
|
143 |
-
|
144 |
-
tokens = get_file(
|
145 |
-
repo_id=repo_id,
|
146 |
-
filename="tokens.txt",
|
147 |
-
subfolder=".",
|
148 |
-
)
|
149 |
-
|
150 |
-
tts_config = sherpa_onnx.OfflineTtsConfig(
|
151 |
-
model=sherpa_onnx.OfflineTtsModelConfig(
|
152 |
-
vits=sherpa_onnx.OfflineTtsVitsModelConfig(
|
153 |
-
model=model,
|
154 |
-
lexicon=lexicon,
|
155 |
-
tokens=tokens,
|
156 |
-
length_scale=1.0 / speed,
|
157 |
-
),
|
158 |
-
provider="cpu",
|
159 |
-
debug=True,
|
160 |
-
num_threads=2,
|
161 |
-
)
|
162 |
-
)
|
163 |
-
tts = sherpa_onnx.OfflineTts(tts_config)
|
164 |
-
|
165 |
-
return tts
|
166 |
-
|
167 |
-
|
168 |
-
@lru_cache(maxsize=10)
|
169 |
-
def _get_vits_piper_en_US(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
170 |
n = len("vits-piper-")
|
171 |
name = repo_id.split("/")[1][n:]
|
172 |
|
@@ -264,36 +212,36 @@ chinese_models = {
|
|
264 |
|
265 |
english_models = {
|
266 |
"csukuangfj/vits-vctk": _get_vits_vctk,
|
267 |
-
"csukuangfj/vits-piper-en_US-amy-low":
|
268 |
-
"csukuangfj/vits-piper-en_US-amy-medium":
|
269 |
-
"csukuangfj/vits-piper-en_US-arctic-medium":
|
270 |
-
"csukuangfj/vits-piper-en_US-danny-low":
|
271 |
-
"csukuangfj/vits-piper-en_US-hfc_male-medium":
|
272 |
-
"csukuangfj/vits-piper-en_US-joe-medium":
|
273 |
-
"csukuangfj/vits-piper-en_US-kathleen-low":
|
274 |
-
"csukuangfj/vits-piper-en_US-kusal-medium":
|
275 |
-
"csukuangfj/vits-piper-en_US-l2arctic-medium":
|
276 |
-
"csukuangfj/vits-piper-en_US-lessac-low":
|
277 |
-
"csukuangfj/vits-piper-en_US-lessac-medium":
|
278 |
-
"csukuangfj/vits-piper-en_US-lessac-high":
|
279 |
-
"csukuangfj/vits-piper-en_US-libritts-high":
|
280 |
-
"csukuangfj/vits-piper-en_US-libritts_r-medium":
|
281 |
-
"csukuangfj/vits-piper-en_US-ryan-low":
|
282 |
-
"csukuangfj/vits-piper-en_US-ryan-medium":
|
283 |
-
"csukuangfj/vits-piper-en_US-ryan-high":
|
284 |
"csukuangfj/vits-ljs": _get_vits_ljs,
|
285 |
}
|
286 |
|
287 |
german_models = {
|
288 |
-
"csukuangfj/vits-piper-de_DE-eva_k-x_low":
|
289 |
-
"csukuangfj/vits-piper-de_DE-karlsson-low":
|
290 |
-
"csukuangfj/vits-piper-de_DE-kerstin-low":
|
291 |
-
"csukuangfj/vits-piper-de_DE-pavoque-low":
|
292 |
-
"csukuangfj/vits-piper-de_DE-ramona-low":
|
293 |
-
"csukuangfj/vits-piper-de_DE-thorsten-low":
|
294 |
-
"csukuangfj/vits-piper-de_DE-thorsten-medium":
|
295 |
-
"csukuangfj/vits-piper-de_DE-thorsten-high":
|
296 |
-
"csukuangfj/vits-piper-de_DE-thorsten_emotional-medium":
|
297 |
}
|
298 |
|
299 |
|
|
|
114 |
|
115 |
|
116 |
@lru_cache(maxsize=10)
|
117 |
+
def _get_vits_piper(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
118 |
n = len("vits-piper-")
|
119 |
name = repo_id.split("/")[1][n:]
|
120 |
|
|
|
212 |
|
213 |
english_models = {
|
214 |
"csukuangfj/vits-vctk": _get_vits_vctk,
|
215 |
+
"csukuangfj/vits-piper-en_US-amy-low": _get_vits_piper,
|
216 |
+
"csukuangfj/vits-piper-en_US-amy-medium": _get_vits_piper,
|
217 |
+
"csukuangfj/vits-piper-en_US-arctic-medium": _get_vits_piper,
|
218 |
+
"csukuangfj/vits-piper-en_US-danny-low": _get_vits_piper,
|
219 |
+
"csukuangfj/vits-piper-en_US-hfc_male-medium": _get_vits_piper,
|
220 |
+
"csukuangfj/vits-piper-en_US-joe-medium": _get_vits_piper,
|
221 |
+
"csukuangfj/vits-piper-en_US-kathleen-low": _get_vits_piper,
|
222 |
+
"csukuangfj/vits-piper-en_US-kusal-medium": _get_vits_piper,
|
223 |
+
"csukuangfj/vits-piper-en_US-l2arctic-medium": _get_vits_piper,
|
224 |
+
"csukuangfj/vits-piper-en_US-lessac-low": _get_vits_piper,
|
225 |
+
"csukuangfj/vits-piper-en_US-lessac-medium": _get_vits_piper,
|
226 |
+
"csukuangfj/vits-piper-en_US-lessac-high": _get_vits_piper,
|
227 |
+
"csukuangfj/vits-piper-en_US-libritts-high": _get_vits_piper,
|
228 |
+
"csukuangfj/vits-piper-en_US-libritts_r-medium": _get_vits_piper,
|
229 |
+
"csukuangfj/vits-piper-en_US-ryan-low": _get_vits_piper,
|
230 |
+
"csukuangfj/vits-piper-en_US-ryan-medium": _get_vits_piper,
|
231 |
+
"csukuangfj/vits-piper-en_US-ryan-high": _get_vits_piper,
|
232 |
"csukuangfj/vits-ljs": _get_vits_ljs,
|
233 |
}
|
234 |
|
235 |
german_models = {
|
236 |
+
"csukuangfj/vits-piper-de_DE-eva_k-x_low": _get_vits_piper,
|
237 |
+
"csukuangfj/vits-piper-de_DE-karlsson-low": _get_vits_piper,
|
238 |
+
"csukuangfj/vits-piper-de_DE-kerstin-low": _get_vits_piper,
|
239 |
+
"csukuangfj/vits-piper-de_DE-pavoque-low": _get_vits_piper,
|
240 |
+
"csukuangfj/vits-piper-de_DE-ramona-low": _get_vits_piper,
|
241 |
+
"csukuangfj/vits-piper-de_DE-thorsten-low": _get_vits_piper,
|
242 |
+
"csukuangfj/vits-piper-de_DE-thorsten-medium": _get_vits_piper,
|
243 |
+
"csukuangfj/vits-piper-de_DE-thorsten-high": _get_vits_piper,
|
244 |
+
"csukuangfj/vits-piper-de_DE-thorsten_emotional-medium": _get_vits_piper,
|
245 |
}
|
246 |
|
247 |
|