requirements.txt

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
#https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu11torch2.2cxx11abiFALSE-cp310-cp310-linux_x86_64.whl