Instructions to use nari-labs/Dia2-2B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Dia2
How to use nari-labs/Dia2-2B with Dia2:
from dia2 import Dia2, GenerationConfig, SamplingConfig dia = Dia2.from_repo("nari-labs/Dia2-2B", 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) - Notebooks
- Google Colab
- Kaggle
Update library tag for better download tracking and code snippets!
#1
by Steveeeeeeen HF Staff - opened
No description provided.
Hey team!
I am Aritra from Hugging Face. I would love to see this PR getting merged. This would help in the following:
- Better download metrics
- Auto generated inference code snippets
- Better search results for models
Let me know if you need any help.
LGTM
NariLabs changed pull request status to merged