MusicGen / requirements.txt
p20523810's picture
Updating packages requirements.txt
79cd30f verified
raw
history blame
333 Bytes
# please make sure you have already a pytorch install that is cuda enabled!
av
einops
flashy
hydra-core==1.3.0
hydra_colorlog
julius
num2words
numpy
sentencepiece
spacy==3.6.0
torch
torchaudio
huggingface_hub
tqdm
transformers==4.40.0 # need Encodec there.
xformers
demucs
librosa
gradio_client==0.3.0
torchmetrics
encodec
protobuf