torch==2.1.0
torchdiffeq==0.2.5
torchvision==0.16.0
numpy==1.26.3
diffusers==0.32.1
accelerate==1.2.1
transformers==4.47.1
huggingface-hub==0.25.0 
tensorboard
gradio
click
opencv-python
scikit-image
numba==0.60.0
scipy
tqdm
einops
sentencepiece
hf_xet
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.2.post1/flash_attn-2.7.2.post1+cu12torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl