Metadata-Version: 2.2
Name: diajax
Version: 0.1.2
Summary: Dia-JAX: A JAX port of Dia, the text-to-speech model for generating realistic dialogue from text with emotion and tone control
Home-page: https://github.com/jaco-bro/diajax
Author: J Joe
Author-email: backupjjoe@gmail.com
License: Apache-2.0
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: jax-metal==0.1.1; platform_system == "Darwin"
Requires-Dist: jaxlib==0.5.0; platform_system == "Darwin"
Requires-Dist: jax==0.5.0; platform_system == "Darwin"
Requires-Dist: flax==0.10.4
Requires-Dist: safetensors
Requires-Dist: pydantic
Requires-Dist: huggingface-hub
Requires-Dist: soundfile
Requires-Dist: descript-audio-codec
Requires-Dist: torchaudio
Provides-Extra: gpu
Requires-Dist: jax[cuda12]; extra == "gpu"
Provides-Extra: tpu
Requires-Dist: jax[tpu]; extra == "tpu"
Dynamic: author
Dynamic: author-email
Dynamic: description
Dynamic: description-content-type
Dynamic: home-page
Dynamic: license
Dynamic: provides-extra
Dynamic: requires-dist
Dynamic: summary

# Dia-JAX

[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/jaco-bro/diajax/blob/main/assets/diajax_TPU.ipynb)

**An experimental JAX port of Dia, the 1.6B parameter text-to-speech model from Nari Labs**

## Quickstart

[output.mp3](https://raw.githubusercontent.com/jaco-bro/diajax/main/assets/example_output.mp3)

```bash
pip install -U diajax
dia --text "[S1] Dear Jacks, to generate audio from text from any machine. [S2] Any machine? (gasps) How? [S1] With flakes and an axe. (chuckle) " --max-tokens=600
```

```python
import diajax
model = diajax.load()
output = diajax.generate(model, "[S1] Dear Jacks, to generate audio from text from any machine. [S2] Any machine? (laughs) How? [S1] With flacks and an axe. (coughs)")
diajax.save(output)
```

## Acknowledgments

This project is a port of the [original Dia model](https://github.com/nari-labs/dia) by Nari Labs. We thank them for releasing their model and code, which made this port possible.

## License

This project is licensed under the same terms as the original Dia model. See [LICENSE](LICENSE) for details.
