ai-tube-model-musicgen-4 / requirements.txt
jbilcke-hf's picture
jbilcke-hf HF staff
Update requirements.txt
8cd6f60 verified
raw
history blame
372 Bytes
# 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==1.26.4
sentencepiece
spacy==3.5.2
torch<2.1.0
torchaudio<2.1.0
huggingface_hub
tqdm
transformers>=4.31.0 # need Encodec there.
xformers==0.0.22
demucs
librosa
gradio_client==0.2.6
torchmetrics
encodec
protobuf
pydub