OpenSUNO / requirements.txt
ginipick's picture
Update requirements.txt
6c32331 verified
raw
history blame
679 Bytes
torch==2.2.0
torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu118
omegaconf>=2.3.0
einops>=0.6.1
numpy<2
transformers>=4.31.0
sentencepiece>=0.1.99
tqdm>=4.65.0
tensorboard>=2.13.0
descript-audiotools>=0.7.2
descript-audio-codec>=0.3.0
scipy==1.10.1
huggingface-hub==0.25.2
wheel>=0.40.0
gradio>=4.0.0
pandas>=2.0.0
librosa>=0.10.0
soundfile>=0.12.1
ffmpeg-python>=0.2.0
accelerate>=0.21.0
bitsandbytes>=0.41.0
safetensors>=0.3.1
tokenizers>=0.13.3
pyyaml>=6.0
regex>=2023.6.3
requests>=2.31.0
typing_extensions>=4.5.0
#https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu11torch2.2cxx11abiFALSE-cp310-cp310-linux_x86_64.whlhl