requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu117
torch
https://download.pytorch.org/whl/cu117/torch-1.13.1%2Bcu117-cp38-cp38-linux_x86_64.whl
torchvision==0.14.1+cu117
#diffusers
git+https://github.com/huggingface/diffusers.git
#transformers
git+https://github.com/huggingface/transformers
scipy
ftfy
psutil
accelerate==0.16.0
OmegaConf
pytorch_lightning
triton==2.0.0.dev20230208
#https://github.com/apolinario/xformers/releases/download/0.0.3/xformers-0.0.14.dev0-cp38-cp38-linux_x86_64.whl
https://github.com/ZyCromerZ/xformers_builds/releases/download/xformers-2023-02-12-Py-3.8-Cuda-11.8.0-PyTorch-1.13.1%2Bcu117-ubuntu-18.04/xformers-0.0.17+12c076d.d20230212-cp38-cp38-linux_x86_64.whl