Skip to main content

Dia-JAX

Open In Colab

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

Quickstart

output.mp3

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
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 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 for details.

Metadata

Release files for diajax 0.1.7

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for diajax 0.1.7
File Size Uploaded
diajax-0.1.7.tar.gz 17.8 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for diajax 0.1.7
File Interpreter ABI Platform
diajax-0.1.7-py3-none-any.whl Python 3 none any Details

Total release size: 35.3 kB

Release files / diajax-0.1.7.tar.gz

Download URL diajax-0.1.7.tar.gz
Size 17.8 kB
Tags Source
SHA-256 checksum
How to use checksums
7ae86798c4870c7d90a70f8229c42b2014f22a58a3273519cae184cbecde5e0a
BLAKE2b-256 checksum
How to use checksums
c8e98999003fa49af6bf7216dedf5e9bdc2e9d92f727d41f3b05db61adfd046f
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.1.0 CPython/3.12.8

Release files / diajax-0.1.7-py3-none-any.whl

Download URL diajax-0.1.7-py3-none-any.whl
Size 17.5 kB
Tags Python 3
SHA-256 checksum
How to use checksums
0b828e743154d003b4f4d1aa5cf3e0e0e8db19de9fb630fa93faf4964b4a97e4
BLAKE2b-256 checksum
How to use checksums
ad4fe187b54dbbf5b03187a26d95b79dba44e9be1193232d7f7067661e0408f0
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.1.0 CPython/3.12.8
Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page