安装 SongGeneration 需要以下步骤:
- 创建 Python 环境:使用 conda 创建虚拟环境,推荐 Python 3.8.12 或更高版本。
- 安装依赖:包括 PyTorch 和 FFmpeg,可通过 pip 安装。
- 安装 Flash Attention(可选):用于加速推理,需要 CUDA 11.8 或以上。
- 下载模型权重:从 Hugging Face 下载模型权重和配置文件,保存至项目根目录。
- Docker 安装(可选):使用官方 Docker 镜像简化配置。
详细安装流程可参考官方 GitHub 仓库的说明。
本答案来源于文章《SongGeneration:生成高品质音乐和歌词的开源AI模型》