dreambooth-training / requirements-local.txt
multimodalart's picture
v2 support
8d49336
raw
history blame
371 Bytes
--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
pip install git+https://github.com/facebookresearch/xformers@7e4c02c#egg=xformers