nazdridoy commited on
Commit
79662c9
·
verified ·
1 Parent(s): a6a746c

feat(tts): add nari-labs/Dia-1.6B model support

Browse files

- [feat] Add "nari-labs/Dia-1.6B" to TTS model choices (ui_components.py:447)
- [feat] Implement visibility logic for "nari-labs/Dia-1.6B" model (ui_components.py:520)
- [ui] Adjust default TTS UI element visibility in `else` block (ui_components.py:522)

Files changed (1) hide show
  1. ui_components.py +4 -2
ui_components.py CHANGED
@@ -445,7 +445,7 @@ def create_tts_tab(handle_tts_generation_fn):
445
  with gr.Group():
446
  gr.Markdown("**🤖 Model & Provider**")
447
  tts_model_name = gr.Dropdown(
448
- choices=["hexgrad/Kokoro-82M", "ResembleAI/chatterbox"],
449
  value=DEFAULT_TTS_MODEL,
450
  label="Model",
451
  info="Select TTS model"
@@ -518,8 +518,10 @@ def create_tts_tab(handle_tts_generation_fn):
518
  return gr.update(visible=True), gr.update(visible=False)
519
  elif model_name == "ResembleAI/chatterbox":
520
  return gr.update(visible=False), gr.update(visible=True)
 
 
521
  else:
522
- return gr.update(visible=True), gr.update(visible=False)
523
 
524
  # Connect model change event
525
  tts_model_name.change(
 
445
  with gr.Group():
446
  gr.Markdown("**🤖 Model & Provider**")
447
  tts_model_name = gr.Dropdown(
448
+ choices=["hexgrad/Kokoro-82M", "ResembleAI/chatterbox", "nari-labs/Dia-1.6B"],
449
  value=DEFAULT_TTS_MODEL,
450
  label="Model",
451
  info="Select TTS model"
 
518
  return gr.update(visible=True), gr.update(visible=False)
519
  elif model_name == "ResembleAI/chatterbox":
520
  return gr.update(visible=False), gr.update(visible=True)
521
+ elif model_name == "nari-labs/Dia-1.6B":
522
+ return gr.update(visible=False), gr.update(visible=False)
523
  else:
524
+ return gr.update(visible=False), gr.update(visible=False)
525
 
526
  # Connect model change event
527
  tts_model_name.change(