Views
No views yet
1conda create -n SongBloom python==3.8.12
2conda activate SongBloom
3
4# yum install libsndfile
5# pip install torch==2.2.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu118 # For different CUDA version
6pip install -r requirements.txt1{
2 "idx": "The index of each sample",
3 "lyrics": "The lyrics to be generated",
4 "prompt_wav": "The path of the style prompt audio",
5}1source set_env.sh
2
3python3 infer.py --input-jsonl example/test.jsonl
4
5# For GPUs with low VRAM like RTX4090, you should set the dtype as bfloat16
6python3 infer.py --input-jsonl example/test.jsonl --dtype bfloat16
7
8# SongBloom also supports flash-attn (optional). To enable it, please install flash-attn (v2.6.3 is used during training) manually and set os.environ['DISABLE_FLASH_ATTN'] = "0" in infer.py:8| Name | Size | Max Length | Prompt type | 🤗 |
|---|---|---|---|---|
| songbloom_full_150s | 2B | 2m30s | 10s wav | link |
| ... |
@article{yang2025songbloom,
title={SongBloom: Coherent Song Generation via Interleaved Autoregressive Sketching and Diffusion Refinement},
author={Yang, Chenyu and Wang, Shuai and Chen, Hangting and Tan, Wei and Yu, Jianwei and Li, Haizhou},
journal={arXiv preprint arXiv:2506.07634},
year={2025}
}