File size: 883 Bytes
500cbb0 |
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 32 33 34 |
---
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
```
```python
from qa_mdt.pipeline import MOSDiffusionPipeline
pipe = MOSDiffusionPipeline(
config_yaml="qa_mdt/audioldm_train/config/mos_as_token/qa_mdt.yaml",
list_inference="qa_mdt/test_prompts/good_prompts_1.lst",
reload_from_ckpt="qamdt/checkpoints/checkpoint_389999.ckpt",
base_folder=None
)
pipe()
```
|