protogen-web-ui / requirements.txt
darkstorm2150's picture
Update requirements.txt
db61d94
raw history blame
No virus
625 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 is broken due transition
triton==2.0.0a2
#https://github.com/apolinario/xformers/releases/download/0.0.3/xformers-0.0.14.dev0-cp38-cp38-linux_x86_64.whl
https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.15/xformers-0.0.15.dev0+4c06c79.d20221205-cp38-cp38-linux_x86_64.whl
tensorflow>=2.10.0