YuE-music-generator-demo / requirements.txt
KingNish's picture
Update requirements.txt
ce09c80 verified
raw
history blame contribute delete
274 Bytes
torch==2.2.0
torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu118
omegaconf
einops
numpy<2
transformers
sentencepiece
tqdm
tensorboard
descript-audiotools>=0.7.2
descript-audio-codec
scipy==1.10.1
huggingface-hub==0.25.2
wheel
accelerate
bitsandbytes>=0.43.3