requirements.txt

accelerate
diffusers
torch
torchvision
transformers
diffusers
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.2.post1/flash_attn-2.7.2.post1+cu11torch2.2cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
xformers
numpy
opencv-python
peft
sentencepiece
pytest
matplotlib
scikit-learn
scipy
einops
jaxtyping
bitsandbytes
beautifulsoup4
git+https://github.com/facebookresearch/pytorch3d.git@stable