From e26cb42e9d94716a254e938cf1b862de77753729 Mon Sep 17 00:00:00 2001 From: Deep-unlearning Date: Mon, 11 Aug 2025 11:18:32 +0200 Subject: [PATCH 1/5] Add KittenML code snippet --- packages/tasks/src/model-libraries-snippets.ts | 11 +++++++++++ packages/tasks/src/model-libraries.ts | 8 ++++++++ 2 files changed, 19 insertions(+) diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index 15e1774fe7..69c9745ec5 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", From 1c59686ae5f3cfe6851aec54e735932d45ed9df0 Mon Sep 17 00:00:00 2001 From: Steven Zheng <58599908+Deep-unlearning@users.noreply.github.com> Date: Tue, 12 Aug 2025 18:26:56 +0200 Subject: [PATCH 2/5] Update packages/tasks/src/model-libraries-snippets.ts Co-authored-by: Lucain --- packages/tasks/src/model-libraries-snippets.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index 69c9745ec5..746c009535 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -849,7 +849,7 @@ 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)` +sf.write('output.wav', audio, 24000)`, ]; export const lightning_ir = (model: ModelData): string[] => { From c8625adc77151e176bdb979b6bc65ea83e4eec17 Mon Sep 17 00:00:00 2001 From: Lucain Date: Thu, 28 Aug 2025 12:11:25 +0200 Subject: [PATCH 3/5] Update packages/tasks/src/model-libraries.ts --- packages/tasks/src/model-libraries.ts | 2 -- 1 file changed, 2 deletions(-) diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 00782b7bb4..26688047cd 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -529,8 +529,6 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { repoName: "KittenTTS", repoUrl: "https://github.com/KittenML/KittenTTS", snippets: snippets.kittentts, - filter: false, - countDownloads: `path:"config.json"`, }, kronos: { prettyLabel: "KRONOS", From 3bdbf42c31ec325716090799ad2ab52d492778a2 Mon Sep 17 00:00:00 2001 From: Deep-unlearning Date: Thu, 28 Aug 2025 14:20:14 +0200 Subject: [PATCH 4/5] changed library name --- packages/tasks/src/model-libraries.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 14e616fcf3..3752c919c4 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -533,7 +533,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { }, kittentts: { prettyLabel: "KittenTTS", - repoName: "KittenTTS", + repoName: "kittentts", repoUrl: "https://github.com/KittenML/KittenTTS", snippets: snippets.kittentts, }, From f3cdd912501920c20233e898984ac52993bebc5b Mon Sep 17 00:00:00 2001 From: Steven Zheng <58599908+Deep-unlearning@users.noreply.github.com> Date: Thu, 28 Aug 2025 14:32:46 +0200 Subject: [PATCH 5/5] Update packages/tasks/src/model-libraries.ts Co-authored-by: Lucain --- packages/tasks/src/model-libraries.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 3752c919c4..14e616fcf3 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -533,7 +533,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { }, kittentts: { prettyLabel: "KittenTTS", - repoName: "kittentts", + repoName: "KittenTTS", repoUrl: "https://github.com/KittenML/KittenTTS", snippets: snippets.kittentts, },