Update requirements.txt

requirements.txt

@@ -1,28 +1,30 @@
-# please make sure you have already a pytorch install that is cuda enabled!
-av==11.0.0
-einops
-flashy>=0.0.1
-hydra-core>=1.1
-hydra_colorlog
-julius
-num2words
-numpy<2.0.0
-sentencepiece
-spacy==3.7.6
-torch==2.1.0
-torchaudio>=2.0.0,<2.1.2
-huggingface_hub
-tqdm
-transformers>=4.31.0  # need Encodec there.
-xformers<0.0.23
-demucs
-librosa
-soundfile
-gradio
-torchmetrics
-encodec
-protobuf
-torchvision==0.16.0
-torchtext==0.16.0
-pesq
-pystoi
+# please make sure you have already a pytorch install that is cuda enabled!
+git+https://github.com/facebookresearch/audiocraft.git
+
+av==11.0.0
+einops
+flashy>=0.0.1
+hydra-core>=1.1
+hydra_colorlog
+julius
+num2words
+numpy<2.0.0
+sentencepiece
+spacy==3.7.6
+torch==2.1.0
+torchaudio>=2.0.0,<2.1.2
+huggingface_hub
+tqdm
+transformers>=4.31.0  # need Encodec there.
+xformers<0.0.23
+demucs
+librosa
+soundfile
+gradio
+torchmetrics
+encodec
+protobuf
+torchvision==0.16.0
+torchtext==0.16.0
+pesq
+pystoi