diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index 15e1774fe7..746c009535 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -841,6 +841,17 @@ print(text) `, ]; +export const kittentts = (model: ModelData): string[] => [ + `from kittentts import KittenTTS +m = KittenTTS("${model.id}") + +audio = m.generate("This high quality TTS model works without a GPU") + +# Save the audio +import soundfile as sf +sf.write('output.wav', audio, 24000)`, +]; + export const lightning_ir = (model: ModelData): string[] => { if (model.tags.includes("bi-encoder")) { return [ diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index d267b199dd..00782b7bb4 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -524,6 +524,14 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { snippets: snippets.kimi_audio, filter: false, }, + kittentts: { + prettyLabel: "KittenTTS", + repoName: "KittenTTS", + repoUrl: "https://github.com/KittenML/KittenTTS", + snippets: snippets.kittentts, + filter: false, + countDownloads: `path:"config.json"`, + }, kronos: { prettyLabel: "KRONOS", repoName: "KRONOS",