dreambooth-training / requirements.txt
multimodalart's picture
Fix locally and let it work with A10G
c74a0ea
raw
history blame
321 Bytes
--extra-index-url https://download.pytorch.org/whl/cu113
torch==1.12.1+cu113
torchvision==0.13.1+cu113
git+https://github.com/huggingface/diffusers.git
accelerate==0.12.0
OmegaConf
wget
pytorch_lightning
huggingface_hub
ftfy
transformers
pyfiglet
triton==2.0.0.dev20220701
bitsandbytes
python-slugify
requests
tensorboard