torch==2.1.0
torchvision==0.16.0
diffusers
tensorboard
transformers
gradio
torchdiffeq
click
torchvision
opencv-python
scikit-image
numba
scipy
tqdm
einops
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.2.post1/flash_attn-2.7.2.post1+cu11torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl