UnlimitedMusicGen / requirements.txt
Surn's picture
Add STYLE model with upgrades
907a484
raw
history blame contribute delete
773 Bytes
# please make sure you have already a pytorch install that is cuda enabled!
av==11.0.0
einops
flashy>=0.0.1
hydra-core>=1.1
hydra_colorlog
torch==2.6.0 --extra-index-url https://download.pytorch.org/whl/cu124
torchaudio>=2.0.0,<2.6.2 --extra-index-url https://download.pytorch.org/whl/cu124
soundfile
huggingface_hub
hf_xet
tqdm
transformers==4.43.4 # need Encodec there.
xformers>=0.0.23 --index-url https://download.pytorch.org/whl/cu124
demucs
librosa==0.11.0
soundfile
gradio[oauth]
pillow
torchdiffeq
torchmetrics
nnAudio
encodec
protobuf>=3.20.1
filetype
wave
mutagen
fastapi>=0.88.0
pydantic
typer
torchvision>=0.21.0 --extra-index-url https://download.pytorch.org/whl/cu124
#torchtext
pesq
pystoi
julius
spacy==3.7.6
sentencepiece
num2words
numpy<1.26.4
matplotlib