安装SongGeneration需要以下环境准备:
- Python环境:建议使用Python 3.8.12或更高版本,可以通过conda创建虚拟环境
- 系统依赖:在Linux系统下需要安装FFmpeg(
yum install ffmpeg
) - Python依赖:使用
pip install -r requirements.txt
安装PyTorch等必要依赖 - 可选项:为提升性能可安装Flash Attention(需要CUDA 11.8或以上版本)
- 模型权重:从Hugging Face下载官方提供的模型权重文件
Windows用户可使用ComfyUI版本简化安装流程,适合非专业开发者使用。
本答案来源于文章《SongGeneration:生成高品质音乐和歌词的开源AI模型》