Skip to content

Commit 877465a

Browse files
authored
Merge branch 'main' into perf-xet-fetchdata
2 parents 699b51a + 5ff0978 commit 877465a

File tree

2 files changed

+17
-0
lines changed

2 files changed

+17
-0
lines changed

packages/tasks/src/model-libraries-snippets.ts

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -841,6 +841,17 @@ print(text)
841841
`,
842842
];
843843

844+
export const kittentts = (model: ModelData): string[] => [
845+
`from kittentts import KittenTTS
846+
m = KittenTTS("${model.id}")
847+
848+
audio = m.generate("This high quality TTS model works without a GPU")
849+
850+
# Save the audio
851+
import soundfile as sf
852+
sf.write('output.wav', audio, 24000)`,
853+
];
854+
844855
export const lightning_ir = (model: ModelData): string[] => {
845856
if (model.tags.includes("bi-encoder")) {
846857
return [

packages/tasks/src/model-libraries.ts

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -531,6 +531,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
531531
snippets: snippets.kimi_audio,
532532
filter: false,
533533
},
534+
kittentts: {
535+
prettyLabel: "KittenTTS",
536+
repoName: "KittenTTS",
537+
repoUrl: "https://github.com/KittenML/KittenTTS",
538+
snippets: snippets.kittentts,
539+
},
534540
kronos: {
535541
prettyLabel: "KRONOS",
536542
repoName: "KRONOS",

0 commit comments

Comments
 (0)