Michael Hu commited on
Commit
6034fea
·
1 Parent(s): 58d9769

set correct fallback sequence

Browse files
Files changed (1) hide show
  1. 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