Skip to content

Commit b31e9d1

Browse files
authored
[Dia] Change ckpt path in docs (#39181)
fix ckpt path
1 parent 18e0cae commit b31e9d1

File tree

1 file changed

+3
-3
lines changed
  • docs/source/en/model_doc

1 file changed

+3
-3
lines changed

docs/source/en/model_doc/dia.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ tokens and decodes them back into audio.
4444
from transformers import AutoProcessor, DiaForConditionalGeneration
4545

4646
torch_device = "cuda"
47-
model_checkpoint = "buttercrab/dia-v1-1.6b"
47+
model_checkpoint = "nari-labs/Dia-1.6B-0626"
4848

4949
text = ["[S1] Dia is an open weights text to dialogue model."]
5050
processor = AutoProcessor.from_pretrained(model_checkpoint)
@@ -66,7 +66,7 @@ from datasets import load_dataset, Audio
6666
from transformers import AutoProcessor, DiaForConditionalGeneration
6767

6868
torch_device = "cuda"
69-
model_checkpoint = "buttercrab/dia-v1-1.6b"
69+
model_checkpoint = "nari-labs/Dia-1.6B-0626"
7070

7171
ds = load_dataset("hf-internal-testing/dailytalk-dummy", split="train")
7272
ds = ds.cast_column("audio", Audio(sampling_rate=44100))
@@ -93,7 +93,7 @@ from datasets import load_dataset, Audio
9393
from transformers import AutoProcessor, DiaForConditionalGeneration
9494

9595
torch_device = "cuda"
96-
model_checkpoint = "buttercrab/dia-v1-1.6b"
96+
model_checkpoint = "nari-labs/Dia-1.6B-0626"
9797

9898
ds = load_dataset("hf-internal-testing/dailytalk-dummy", split="train")
9999
ds = ds.cast_column("audio", Audio(sampling_rate=44100))

0 commit comments

Comments
 (0)