File size: 828 Bytes
49d4029 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | ---
base_model: facebook/musicgen-small
library_name: transformers
pipeline_tag: text-to-audio
tags:
- lora
- musicgen
- music-generation
- peft
---
# MusicGen LoRA Fine-Tuned
LoRA adapter fine-tuned on custom music dataset using `facebook/musicgen-small`.
## Como usar
```python
from transformers import AutoProcessor, MusicgenForConditionalGeneration
from peft import PeftModel
import torch, soundfile as sf
processor = AutoProcessor.from_pretrained("facebook/musicgen-small")
model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")
model.decoder = PeftModel.from_pretrained(model.decoder, "SEU_REPO_AQUI")
inputs = processor(text=["upbeat electronic music"], return_tensors="pt")
audio = model.generate(**inputs, max_new_tokens=256)
sf.write("output.wav", audio[0].cpu().numpy(), 32000)
```
|