File size: 612 Bytes
bf277fe |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import torch
from dia.model import Dia
# Select device: CPU
device = torch.device("cpu")
print(f"Using device: {device}")
# Load model
model = Dia.from_pretrained(
"nari-labs/Dia-1.6B-0626", compute_dtype="float32", device=device
) # Float32 works better than float16 on CPU - you can also test with float16
text = "[S1] Dia is an open weights text to dialogue model. [S2] You get full control over scripts and voices. [S1] Wow. Amazing. (laughs) [S2] Try it now on Git hub or Hugging Face."
output = model.generate(text, use_torch_compile=False, verbose=True)
model.save_audio("simple.mp3", output)
|