finetuned_diffusion / requirements.txt
anzorq's picture
Update requirements.txt
6efc8bf
raw
history blame
428 Bytes
--extra-index-url https://download.pytorch.org/whl/cu113
torch
torchvision==0.13.1+cu113
#diffusers
git+https://github.com/huggingface/diffusers.git
#transformers
git+https://github.com/huggingface/transformers
scipy
ftfy
psutil
accelerate==0.12.0
#OmegaConf
#pytorch_lightning
triton==2.0.0.dev20220701
#bitsandbytes
https://github.com/apolinario/xformers/releases/download/0.0.3/xformers-0.0.14.dev0-cp38-cp38-linux_x86_64.whl