mirror of
https://github.com/hanzoai/koe.git
synced 2026-10-30 18:33:00 +00:00
Ammar A.andGitHub
87e7cc365b
* Refactor simple-mac.py to remove device selection and use float16 for model loading * Refactor README.md to simplify model loading for MacOS by removing device selection and using float16
13 lines
566 B
Python
13 lines
566 B
Python
from dia.model import Dia
|
|
|
|
|
|
model = Dia.from_pretrained("nari-labs/Dia-1.6B", compute_dtype="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."
|
|
|
|
# It is important to set the `use_torch_compile` argument to `False` when using Dia on MacOS.
|
|
# This is because the `torch.compile` function is not supported on MacOS.
|
|
output = model.generate(text, use_torch_compile=False, verbose=True)
|
|
|
|
model.save_audio("simple.mp3", output)
|