Spaces:
Sleeping
Sleeping
Michael Hu
commited on
Commit
·
6034fea
1
Parent(s):
58d9769
set correct fallback sequence
Browse files- utils/tts_engines.py +9 -1
utils/tts_engines.py
CHANGED
|
@@ -370,6 +370,9 @@ def get_available_engines() -> List[str]:
|
|
| 370 |
if DIA_AVAILABLE:
|
| 371 |
available.append('dia')
|
| 372 |
|
|
|
|
|
|
|
|
|
|
| 373 |
# Dummy is always available
|
| 374 |
available.append('dummy')
|
| 375 |
|
|
@@ -380,7 +383,7 @@ def create_engine(engine_type: str, lang_code: str = 'z') -> TTSEngineBase:
|
|
| 380 |
"""Create a specific TTS engine
|
| 381 |
|
| 382 |
Args:
|
| 383 |
-
engine_type (str): Type of engine to create ('kokoro', 'kokoro_space', 'dia', 'dummy')
|
| 384 |
lang_code (str): Language code for the engine
|
| 385 |
|
| 386 |
Returns:
|
|
@@ -404,6 +407,11 @@ def create_engine(engine_type: str, lang_code: str = 'z') -> TTSEngineBase:
|
|
| 404 |
raise ValueError("Dia TTS engine is not available")
|
| 405 |
return DiaTTSEngine(lang_code)
|
| 406 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 407 |
elif engine_type == 'dummy':
|
| 408 |
return DummyTTSEngine(lang_code)
|
| 409 |
|
|
|
|
| 370 |
if DIA_AVAILABLE:
|
| 371 |
available.append('dia')
|
| 372 |
|
| 373 |
+
if DIA_SPACE_AVAILABLE:
|
| 374 |
+
available.append('dia_space')
|
| 375 |
+
|
| 376 |
# Dummy is always available
|
| 377 |
available.append('dummy')
|
| 378 |
|
|
|
|
| 383 |
"""Create a specific TTS engine
|
| 384 |
|
| 385 |
Args:
|
| 386 |
+
engine_type (str): Type of engine to create ('kokoro', 'kokoro_space', 'dia', 'dia_space', 'dummy')
|
| 387 |
lang_code (str): Language code for the engine
|
| 388 |
|
| 389 |
Returns:
|
|
|
|
| 407 |
raise ValueError("Dia TTS engine is not available")
|
| 408 |
return DiaTTSEngine(lang_code)
|
| 409 |
|
| 410 |
+
elif engine_type == 'dia_space':
|
| 411 |
+
if not DIA_SPACE_AVAILABLE:
|
| 412 |
+
raise ValueError("Dia Space TTS engine is not available")
|
| 413 |
+
return DiaSpaceTTSEngine(lang_code)
|
| 414 |
+
|
| 415 |
elif engine_type == 'dummy':
|
| 416 |
return DummyTTSEngine(lang_code)
|
| 417 |
|