requirements.txt

torch        
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
diffusers
fairscale
accelerate
tensorboard
transformers
gradio
torchdiffeq
click
h5py
bitsandbytes
torchvision
fastparquet