基础模型：facebook/musicgen-small
库名称：transformers.js
许可证：cc-by-nc-4.0

此版本为适配Transformers.js的ONNX权重版本，原模型地址：https://huggingface.co/facebook/musicgen-small

使用说明（Transformers.js）

若尚未安装，可通过NPM安装Transformers.js库：

npm install @huggingface/transformers

示例： 使用Xenova/musicgen-small生成音乐

import { AutoTokenizer, MusicgenForConditionalGeneration } from '@huggingface/transformers';

// 加载分词器与模型
const tokenizer = await AutoTokenizer.from_pretrained('Xenova/musicgen-small');
const model = await MusicgenForConditionalGeneration.from_pretrained('Xenova/musicgen-small', {
  dtype: {
    text_encoder: 'q8',
    decoder_model_merged: 'q8',
    encodec_decode: 'fp32',
  },
});

// 准备文本输入
const prompt = '一首轻快活泼的电子舞曲，带有切分鼓点、空灵合成音效和强烈情感，BPM：130';
const inputs = tokenizer(prompt);

// 生成音频
const audio_values = await model.generate({
  ...inputs,
  max_new_tokens: 500,
  do_sample: true,
  guidance_scale: 3,
});

// （可选）将输出写入WAV文件
const audio = new RawAudio(audio_values.data, model.config.audio_encoder.sampling_rate);
audio.save('musicgen.wav');

我们还发布了在线演示，欢迎体验：https://huggingface.co/spaces/Xenova/musicgen-web

注：当前为临时方案，在WebML技术普及前需单独存放ONNX权重。若需将模型适配网页端，建议使用🤗 Optimum转换为ONNX格式，并参照本仓库结构（将ONNX权重存放于onnx子文件夹）。