From 970b1d4908e3605186eef6904cda1991ed591b89 Mon Sep 17 00:00:00 2001 From: Deep-unlearning Date: Tue, 10 Jun 2025 18:10:53 +0200 Subject: [PATCH 1/4] add vui code snippet --- packages/tasks/src/model-libraries-snippets.ts | 16 ++++++++++++++++ packages/tasks/src/model-libraries.ts | 7 +++++++ 2 files changed, 23 insertions(+) diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index dbe316c4b3..20fa84839e 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -1332,6 +1332,22 @@ export const voicecraft = (model: ModelData): string[] => [ model = VoiceCraft.from_pretrained("${model.id}")`, ]; +export const vui = (model: ModelData): string[] => [ + `import torchaudio + +from vui.inference import render +from vui.model import Vui, + +model = Vui.from_pretrained().cuda() +waveform = render( + model, + "Hey, here is some random stuff, usually something quite long as the shorter the text the less likely the model can cope!", +) +print(waveform.shape) +torchaudio.save("out.opus", waveform[0], 22050) +`, +]; + export const chattts = (): string[] => [ `import ChatTTS import torchaudio diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 101c3cc6f4..78dc038b77 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -1022,6 +1022,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { docsUrl: "https://github.com/jasonppy/VoiceCraft", snippets: snippets.voicecraft, }, + vui: { + prettyLabel: "VUI", + repoName: "VUI", + repoUrl: "https://github.com/vui-ai/vui", + countDownloads: `path_extension:"pt"`, + snippets: snippets.vui, + }, wham: { prettyLabel: "WHAM", repoName: "wham", From bce3d396281b7f4d17ddc9f977813412113a045b Mon Sep 17 00:00:00 2001 From: Steven Zheng <58599908+Deep-unlearning@users.noreply.github.com> Date: Wed, 11 Jun 2025 10:58:01 +0200 Subject: [PATCH 2/4] Update packages/tasks/src/model-libraries-snippets.ts Co-authored-by: Lucain --- packages/tasks/src/model-libraries-snippets.ts | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index 20fa84839e..e5a6f4f250 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -1333,7 +1333,8 @@ model = VoiceCraft.from_pretrained("${model.id}")`, ]; export const vui = (model: ModelData): string[] => [ - `import torchaudio + `# !pip install git+https://github.com/fluxions-ai/vui +import torchaudio from vui.inference import render from vui.model import Vui, From ee969b3702a4baea7715290f2ed7a5d6dda8da44 Mon Sep 17 00:00:00 2001 From: Steven Zheng <58599908+Deep-unlearning@users.noreply.github.com> Date: Wed, 11 Jun 2025 10:58:31 +0200 Subject: [PATCH 3/4] Update packages/tasks/src/model-libraries.ts Co-authored-by: Lucain --- packages/tasks/src/model-libraries.ts | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 78dc038b77..fab7fe479f 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -1023,8 +1023,8 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { snippets: snippets.voicecraft, }, vui: { - prettyLabel: "VUI", - repoName: "VUI", + prettyLabel: "Vui", + repoName: "Vui", repoUrl: "https://github.com/vui-ai/vui", countDownloads: `path_extension:"pt"`, snippets: snippets.vui, From 2d98317863f1913d6a77abef9c96fc3a2977918b Mon Sep 17 00:00:00 2001 From: Steven Zheng <58599908+Deep-unlearning@users.noreply.github.com> Date: Fri, 13 Jun 2025 11:53:53 +0200 Subject: [PATCH 4/4] Update packages/tasks/src/model-libraries-snippets.ts Co-authored-by: vb --- packages/tasks/src/model-libraries-snippets.ts | 1 + 1 file changed, 1 insertion(+) diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index f2328b5029..9855af3810 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -1356,6 +1356,7 @@ model = VoiceCraft.from_pretrained("${model.id}")`, export const vui = (model: ModelData): string[] => [ `# !pip install git+https://github.com/fluxions-ai/vui + import torchaudio from vui.inference import render