OpenMusic / qa_mdt /README.md
jadechoghari's picture
add model
9b9e0ee verified
|
raw
history blame
No virus
1.1 kB
---
library_name: diffusers
tags:
- music
---
# Hugging Face Diffusers Implementation of QA-MDT
**QADMT: Quality-Aware Diffusion for Text-to-Music 🎶**
QADMT brings a new approach to text-to-music generation by using quality-aware training to tackle issues like low-fidelity audio and weak labeling in datasets.
With a masked diffusion transformer (MDT), QADMT delivers SOTA results on MusicCaps and Song-Describer, enhancing both quality and musicality.
## Usage:
```bash
!git lfs install
!git clone https://huggingface.co/jadechoghari/qa-mdt
```
```bash
pip install -r qa_mdt/requirements.txt
pip install xformers==0.0.26.post1
pip install torchlibrosa==0.0.9 librosa==0.9.2
pip install -q pytorch_lightning==2.1.3 torchlibrosa==0.0.9 librosa==0.9.2 ftfy==6.1.1 braceexpand
pip install torch==2.3.0+cu121 torchvision==0.18.0+cu121 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu121
```
```python
from qa_mdt.pipeline import MOSDiffusionPipeline
pipe = MOSDiffusionPipeline()
pipe("A modern synthesizer creating futuristic soundscapes.")
```
# Enjoy the music!! 🎶