Update README.md

README.md

@@ -20,7 +20,11 @@ With a masked diffusion transformer (MDT), QADMT delivers SOTA results on MusicC
 ```
 
 ```bash
-pip install -r qa_mdt/requirements.txt --index-url https://download.pytorch.org/whl/cu121
+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
@@ -29,3 +33,5 @@ from qa_mdt.pipeline import MOSDiffusionPipeline
 pipe = MOSDiffusionPipeline()
 pipe("A modern synthesizer creating futuristic soundscapes.")
 ```
+
+# Enjoy the music!! 🎶