MusicGen / requirements.txt
ford442's picture
Update requirements.txt
ace46fc verified
raw
history blame contribute delete
830 Bytes
# please make sure you have already a pytorch install that is cuda enabled!
#--extra-index-url https://download.pytorch.org/whl/cu124
--prefer-binary
#torch
#torchvision
#torchaudio
spaces
audiocraft==1.3.0
pydantic #==1.10.21
spacy==4.0.0.dev3
thinc
#torchaudio
pybind11
av
einops
flashy>=0.0.1
hydra-core>=1.1
hydra_colorlog
julius
num2words
numpy<2.0
sentencepiece
#spacy==3.5.2
#torch<2.1.0
#torchaudio<2.1.0
#huggingface_hub
tqdm
#transformers>=4.31.0 # need Encodec there.
#git+https://github.com/huggingface/transformers.git
#git+https://github.com/huggingface/accelerate.git
xformers
demucs
librosa
gradio_client #==0.2.6
torchmetrics
encodec
protobuf
gradio==3.44.4
accelerate==0.24.0
tokenizers==0.14.1
huggingface_hub==0.21.0
diffusers==0.21.4
torchaudio==2.1.0
torchvision==0.16.0
transformers==4.34.0