Spaces:
Running
Running
Commit
·
de8dd8a
1
Parent(s):
edff7b9
add kitten v0.2
Browse files
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 (
|
|
|
|
|
|
|
|
|
|
| 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,
|