Spaces:
Running
Running
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 |
|