File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -44,7 +44,7 @@ tokens and decodes them back into audio.
44
44
from transformers import AutoProcessor, DiaForConditionalGeneration
45
45
46
46
torch_device = " cuda"
47
- model_checkpoint = " buttercrab/dia-v1-1.6b "
47
+ model_checkpoint = " nari-labs/Dia-1.6B-0626 "
48
48
49
49
text = [" [S1] Dia is an open weights text to dialogue model." ]
50
50
processor = AutoProcessor.from_pretrained(model_checkpoint)
@@ -66,7 +66,7 @@ from datasets import load_dataset, Audio
66
66
from transformers import AutoProcessor, DiaForConditionalGeneration
67
67
68
68
torch_device = " cuda"
69
- model_checkpoint = " buttercrab/dia-v1-1.6b "
69
+ model_checkpoint = " nari-labs/Dia-1.6B-0626 "
70
70
71
71
ds = load_dataset(" hf-internal-testing/dailytalk-dummy" , split = " train" )
72
72
ds = ds.cast_column(" audio" , Audio(sampling_rate = 44100 ))
@@ -93,7 +93,7 @@ from datasets import load_dataset, Audio
93
93
from transformers import AutoProcessor, DiaForConditionalGeneration
94
94
95
95
torch_device = " cuda"
96
- model_checkpoint = " buttercrab/dia-v1-1.6b "
96
+ model_checkpoint = " nari-labs/Dia-1.6B-0626 "
97
97
98
98
ds = load_dataset(" hf-internal-testing/dailytalk-dummy" , split = " train" )
99
99
ds = ds.cast_column(" audio" , Audio(sampling_rate = 44100 ))
You can’t perform that action at this time.
0 commit comments