ai-tube-model-musicgen-1 / requirements.txt
patrickvonplaten's picture
Add hf loading & improve a couple of things on the README (#2)
9138f15 unverified
raw
history blame
270 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
sentencepiece
spacy==3.5.2
torch>=2.0.0
torchaudio>=2.0.0
huggingface_hub
tqdm
transformers
xformers
demucs
librosa