Skip to content

Commit 60543d9

Browse files
feat: added support for KokoroTTS in TTSAgent
1 parent 9b4b949 commit 60543d9

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/rai_s2s/rai_s2s/tts/agents/tts_agent.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -140,6 +140,13 @@ def from_config(cls, cfg_path: Optional[str] = None) -> Self:
140140
model = OpenTTS(voice=cfg.text_to_speech.voice)
141141
else:
142142
model = OpenTTS()
143+
case "KokoroTTS":
144+
from rai_s2s.tts.models import KokoroTTS
145+
146+
if cfg.text_to_speech.voice != "":
147+
model = KokoroTTS(voice=cfg.text_to_speech.voice)
148+
else:
149+
model = KokoroTTS()
143150
case _:
144151
raise ValueError(f"Unknown model_type: {cfg.text_to_speech.model_type}")
145152
return cls(config, "rai_auto_tts", model)

0 commit comments

Comments
 (0)