NotaGen 是一個大模型預訓練的音樂生成模型

項目地址：https://github.com/ElectricAlexis/NotaGen

項目DEMO：https://electricalexis.github.io/notagen-demo/

下面試試本地部署效果

首先下載項目文件

git clone https://github.com/ElectricAlexis/NotaGen.git

然后我們創建一個 3.10 的 Python 環境

conda create  --prefix G:\AI\NotaGen python=3.10

然后激活環境

conda activate G:\AI\NotaGen

安裝 pytouch 和 cuda

conda install pytorch==2.3.0 pytorch-cuda=11.8 -c pytorch -c nvidia

安裝 accelerate

pip install accelerate

安裝 optimum

pip install optimum

最后安裝必要的依賴

pip install -r requirements.txt

最后檢查下我們的 pytorch 是不是支持 cuda 的

python
import torch
torch.__version__
print(torch.cuda.is_available())

返回 true 表示支持 cuda，可以使用 GPU

項目自帶了個damo，在 gradio 文件夾中

首先下載模型

demo使用的模型為 NotaGen-X

模型地址：https://huggingface.co/ElectricAlexis/NotaGen/blob/main/weights_notagenx_p_size_16_p_length_1024_p_layers_20_h_size_1280.pth

鏡像站模型地址：https://hf-mirror.com/ElectricAlexis/NotaGen/blob/main/weights_notagenx_p_size_16_p_length_1024_p_layers_20_h_size_1280.pth

然后把模型放到 gradio 文件夾中（主要不要改模型的文件名）

運行Demo：

python demo.py

看下模型已經加載到了顯存中（8G顯存的機器就能跑）

這個時候打開 http://127.0.0.1:7861/ 就能看到 Demo 界面了

隨便選個風格生成下看看，顯卡開始狂轉中…

等待一會，生成完成

保存文件，然后找個在線播放樂譜的網站試試效果，比如：https://bbs.moonyueqi.com/xmlplay/