Dia2 How to use tinytrashlabs/dia2-1b-mlx-bf16 with Dia2:
from dia2 import Dia2, GenerationConfig, SamplingConfig
dia = Dia2.from_repo("tinytrashlabs/dia2-1b-mlx-bf16", 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)
