requirements.txt

--find-links https://storage.googleapis.com/jax/releases/pip/torch_cuda_libs.html
torch
accelerate==1.6.0
transformers==4.51.3
diffusers
tqdm
numpy
scipy
ml-collections
absl-py
gradio
av
aiortc
soundfile
librosa
flash-attn --no-build-isolation