From 7022cbd932fcbe7dbc362e1f2de46ac942793a2b Mon Sep 17 00:00:00 2001 From: Deep-unlearning Date: Tue, 25 Nov 2025 11:07:09 +0100 Subject: [PATCH 1/5] add dia2 snippet --- packages/tasks/src/model-libraries-snippets.ts | 13 +++++++++++++ packages/tasks/src/model-libraries.ts | 7 +++++++ 2 files changed, 20 insertions(+) diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index 15e1774fe7..a6830f3c6c 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -310,6 +310,19 @@ output = model.generate(text) sf.write("simple.mp3", output, 44100)`, ]; +export const dia2 = (model: ModelData): string[] => [ + `from dia2 import Dia2, GenerationConfig, SamplingConfig + +dia = Dia2.from_repo("${model.id}", device="cuda", dtype="bfloat16") +config = GenerationConfig( + cfg_scale=2.0, + audio=SamplingConfig(temperature=0.8, top_k=50), + use_cuda_graph=True, +) +result = dia.generate("[S1] Hello Dia2!", config=config, output_wav="hello.wav", verbose=True) +`, +]; + export const describe_anything = (model: ModelData): string[] => [ `# pip install git+https://github.com/NVlabs/describe-anything from huggingface_hub import snapshot_download diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index d267b199dd..9b0517bda4 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -273,6 +273,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { snippets: snippets.dia, filter: false, }, + "dia2-tts": { + prettyLabel: "Dia2", + repoName: "Dia2", + repoUrl: "https://github.com/nari-labs/dia2", + snippets: snippets.dia2, + filter: false, + }, diffree: { prettyLabel: "Diffree", repoName: "Diffree", From b7a84d9e2b329a7323010c7fe903503968a3647b Mon Sep 17 00:00:00 2001 From: Deep-unlearning Date: Tue, 25 Nov 2025 11:14:56 +0100 Subject: [PATCH 2/5] change to dia2 --- 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 9b0517bda4..b282289edc 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -273,7 +273,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { snippets: snippets.dia, filter: false, }, - "dia2-tts": { + "dia2": { prettyLabel: "Dia2", repoName: "Dia2", repoUrl: "https://github.com/nari-labs/dia2", From 00d1f62df0da332e4c9a8375b6828d615880a1af Mon Sep 17 00:00:00 2001 From: Lucain Date: Tue, 25 Nov 2025 16:10:47 +0100 Subject: [PATCH 3/5] Update packages/tasks/src/model-libraries.ts --- 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 b282289edc..33c8642de3 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -273,7 +273,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { snippets: snippets.dia, filter: false, }, - "dia2": { + dia2: { prettyLabel: "Dia2", repoName: "Dia2", repoUrl: "https://github.com/nari-labs/dia2", From a865440a5d7baf0f8452ccccf9e71283a1433a2e Mon Sep 17 00:00:00 2001 From: Lucain Date: Wed, 26 Nov 2025 16:09:59 +0100 Subject: [PATCH 4/5] Apply suggestion from @Wauplin --- 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 2427b710f4..d6ab1b3fff 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -299,7 +299,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { repoUrl: "https://github.com/nari-labs/dia2", snippets: snippets.dia2, filter: false, - }, + }, "diff-interpretation-tuning": { prettyLabel: "Diff Interpretation Tuning", repoName: "Diff Interpretation Tuning", From c26deb91996d26410277224d445e4e8471bf169e Mon Sep 17 00:00:00 2001 From: Deep-unlearning Date: Thu, 4 Dec 2025 16:17:51 +0100 Subject: [PATCH 5/5] nit --- 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 d6ab1b3fff..3c008db5e1 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -299,7 +299,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { repoUrl: "https://github.com/nari-labs/dia2", snippets: snippets.dia2, filter: false, - }, + }, "diff-interpretation-tuning": { prettyLabel: "Diff Interpretation Tuning", repoName: "Diff Interpretation Tuning",