requirements_audiocraft_deps.txt

torch==2.0.0 # BSD 3-Clause - ensures torch is not reinstalled
xformers==0.0.19 # For torch==2.0.0 project plane
# 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
sentencepiece
spacy>=3.6.1
# torch==2.1.0 # wrong torch version
torchaudio>=2.0.0
huggingface_hub
tqdm
transformers>=4.31.0  # need Encodec there.
xformers
demucs
librosa
gradio
torchmetrics
encodec
protobuf