distilgpt2 abc irish music generation

提供商ehcalabres
分类audio-generation
许可证apache-2.0
下载量153
星标0

简介

这是一个基于 DistilGPT-2 架构微调的轻量级音频生成模型,专注于爱尔兰传统音乐的创作。它将音乐转化为类文本的序列进行学习,能够生成具有爱尔兰民乐风格的旋律。由于采用了蒸馏模型,该模型对计算资源要求极低,非常适合开发者在本地环境快速部署,用于探索符号音乐生成或作为音乐 AI 插件的底层逻辑。对于习惯使用 MIDI 或乐谱生成工具的用户来说,它提供了一种快速生成风格化灵感的新方式。

核心亮点

  • 专注爱尔兰民乐风格,旋律极具地域特色
  • 基于 DistilGPT-2,部署轻量且推理速度快
  • 采用 Apache-2.0 协议,商业化使用无压力
  • 适合作为音乐创作的灵感生成工具

使用方法

安装依赖
# 安装 Hugging Face transformers
pip install transformers torch
SDK 使用
# 使用 transformers 加载模型
from transformers import AutoModel, AutoTokenizer

model = AutoModel.from_pretrained("ehcalabres/distilgpt2-abc-irish-music-generation")
tokenizer = AutoTokenizer.from_pretrained("ehcalabres/distilgpt2-abc-irish-music-generation")

Hugging Face 下载

我们推荐使用命令行或者 Hugging Face Hub SDK 来进行模型的下载。

操作指引:在下载前,请先通过如下命令安装 huggingface_hub:

操作指引
pip install -U huggingface_hub

命令行下载

下载完整模型库

下载完整模型库
huggingface-cli download ehcalabres/distilgpt2-abc-irish-music-generation

下载单个文件到指定本地文件夹(以下载 config.json 到当前路径下 ./dir 目录为例)

下载单个文件到指定本地文件夹(以下载 config.json 到当前路径下 ./dir 目录为例)
huggingface-cli download ehcalabres/distilgpt2-abc-irish-music-generation config.json --local-dir ./dir

更多命令行下载选项,可参见官方文档

SDK 下载

SDK 下载
# 模型下载
from huggingface_hub import snapshot_download
model_dir = snapshot_download('ehcalabres/distilgpt2-abc-irish-music-generation')

Git 下载

请确保 lfs 已经被正确安装

Git 下载
git lfs install
git clone https://huggingface.co/ehcalabres/distilgpt2-abc-irish-music-generation

如果您希望跳过 lfs 大文件下载,可以使用如下命令

跳过 LFS
GIT_LFS_SKIP_SMUDGE=1 git clone https://huggingface.co/ehcalabres/distilgpt2-abc-irish-music-generation

模型文件托管在 Hugging Face Hub,使用 HF CLI / SDK / Git 直接下载,不经过本站。

PyTorch / Transformers 使用

安装 Transformers

安装 Transformers
pip install -U transformers torch

模型加载和推理

模型加载和推理
from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained('ehcalabres/distilgpt2-abc-irish-music-generation')
tokenizer = AutoTokenizer.from_pretrained('ehcalabres/distilgpt2-abc-irish-music-generation')

完整文档

来源: HuggingFace

---
license: apache-2.0
tags:

  • generated_from_trainer

base_model: distilgpt2
model-index:
  • name: distilgpt2-abc-irish-music-generation

results: []
---

<!-- This model card has been generated automatically according to the information the Trainer had access to. You
should probably proofread and complete it, then remove this comment. -->

distilgpt2-abc-irish-music-generation

This model is a fine-tuned version of distilgpt2 on an unknown dataset.

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 5e-05

  • train_batch_size: 4

  • eval_batch_size: 4

  • seed: 42

  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08

  • lr_scheduler_type: linear

  • lr_scheduler_warmup_steps: 500

  • num_epochs: 10

Training results

Framework versions

  • Transformers 4.19.2
  • Pytorch 1.11.0+cu113
  • Datasets 2.2.2
  • Tokenizers 0.12.1