YuE-exllamav2 项目使用教程

1. 项目目录结构及介绍

YuE-exllamav2 项目是一个开源的音乐生成项目，其目录结构如下：

• assets/: 存放项目相关的资源文件，如项目图标等。
• prompt_egs/: 包含一些示例提示文件，用于指导模型生成音乐。
• src/: 源代码目录，包含项目的核心实现代码。
  • yue/: 包含音乐生成的核心逻辑。
• .gitattributes: 定义如何处理特定文件类型的 Git 属性。
• .gitignore: 指定 Git 忽略跟踪的文件和目录。
• LICENSE: 项目许可证文件，本项目采用 Apache-2.0 许可。
• NOTICE: 包含项目的一些法律声明。
• README.md: 项目说明文件，详细介绍项目信息和使用方法。
• requirements-no-wheels.txt: 不包含 wheel 文件的依赖项列表。
• requirements-rocm.txt: ROCm GPU 加速的依赖项列表。
• requirements.txt: 项目依赖项列表。
• top_200_tags.json: 包含项目的一些标签信息。

2. 项目的启动文件介绍

项目的启动文件位于 src/yue/infer.py。该文件是运行音乐生成模型的入口点。使用以下命令运行：

python src/yue/infer.py --stage1_use_exl2 --stage2_use_exl2 --stage2_cache_size 32768 [original args]

其中 [original args] 表示原始的命令行参数。

3. 项目的配置文件介绍

项目的配置文件主要位于 requirements.txt，其中列出了项目运行所需的 Python 包依赖。以下是一些关键的依赖项：

• torch: 用于深度学习的 PyTorch 库。
• torchvision: PyTorch 的视觉库。
• torchaudio: PyTorch 的音频库。
• pip: Python 包管理器。

此外，还有其他一些依赖项，具体请参考 requirements.txt 文件。

在运行项目之前，需要安装这些依赖项。可以使用以下命令：

pip install -r requirements.txt

确保在安装依赖项之前已经创建了 Python 虚拟环境，并且已经激活。

以上就是 YuE-exllamav2 项目的使用教程，按照上述步骤即可开始使用该项目进行音乐生成。