torch==2.1.0
torchaudio==2.1.0
av
julius
omegaconf
einops
num2words
flashy
torchmetrics
sentencepiece
git+https://github.com/facebookresearch/audiocraft.git
xformers==0.0.20 --index-url https://download.pytorch.org/whl/cu118