Songcomposer Sft

🚀 SongComposer

SongComposer 是一款基于 InternLM2 的大语言模型（LLM），专为歌曲创作中的歌词和旋律生成而设计，能够有效解决歌曲创作过程中的创意和效率问题，为音乐创作者提供强大的支持。

🚀 快速开始

从 Transformers 库导入

若要使用 Transformers 库加载 SongComposer_sft 模型，请使用以下代码：

from transformers import AutoTokenizer, AutoModel
ckpt_path = "Mar2Ding/songcomposer_sft"
tokenizer = AutoTokenizer.from_pretrained(ckpt_path, trust_remote_code=True)
model = AutoModel.from_pretrained(ckpt_path, trust_remote_code=True).cuda().half()
prompt = 'Create a song on brave and sacrificing with a rapid pace.'
model.inference(prompt, tokenizer)

通过 Transformers 加载

通过以下代码加载 SongComposer_sft 模型：

from transformers import AutoTokenizer, AutoModel
ckpt_path = "Mar2Ding/songcomposer_sft"
tokenizer = AutoTokenizer.from_pretrained(ckpt_path, trust_remote_code=True)
model = AutoModel.from_pretrained(ckpt_path, trust_remote_code=True).cuda().half()
prompt = 'Create a song on brave and sacrificing with a rapid pace.'
model.inference(prompt, tokenizer)

✨ 主要特性

我们发布了 SongComposer 系列的两个版本：

• SongComposer_pretrain：以 InternLM2 为基础进行预训练的 SongComposer，具备歌词和旋律的基础知识。
• SongComposer_sft：经过微调的 SongComposer，可实现遵循指令的歌曲生成，包括歌词到旋律、旋律到歌词、歌曲续写、文本到歌曲等功能。

📄 许可证

代码采用 Apache-2.0 许可证，模型权重完全开放用于学术研究，也允许免费商业使用。

相关链接

• 💻Github 仓库
• 📖论文