requirements-local.txt

--extra-index-url https://download.pytorch.org/whl/cu113
torch==1.12.1+cu113
torchvision==0.13.1+cu113
diffusers==0.9.0
accelerate==0.12.0
OmegaConf
wget
pytorch_lightning
huggingface_hub
ftfy
transformers
pyfiglet
triton==2.0.0.dev20220701
bitsandbytes
python-slugify
requests
tensorboard
sentencepiece
pip install git+https://github.com/facebookresearch/xformers@7e4c02c#egg=xformers