MusicGen / requirements.txt
mirxa2's picture
Update requirements.txt
fda5477 verified
# please make sure you have already a pytorch install that is cuda enabled!
av
einops
flashy>=0.0.1
hydra-core>=1.1
hydra_colorlog
julius
num2words
numpy<2
sentencepiece
spacy==3.5.2
torch>=2.0.0
torchaudio>=2.0.0
huggingface_hub
tqdm
transformers>=4.31.0
xformers
demucs
librosa
gradio_client==0.2.6
torchmetrics
encodec
protobuf
pydantic<2