mooncast / requirements.txt
jzq11111's picture
Update requirements.txt
397909b verified
torch==2.2.2
torchaudio==2.2.2
sentencepiece==0.2.0
protobuf
librosa==0.9.1
pyyaml
transformers==4.49.0
safetensors
einops
scipy
timm==1.0.7
torchdyn
librosa
accelerate==0.26.0
ninja
cryptography
huggingface_hub
gradio==5.22.0
numpy==1.26.4
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.2cxx11abiFALSE-cp310-cp310-linux_x86_64.whl