Erasing-Concepts-In-Diffusion / requirements.txt
JadenFK's picture
ckpt download as well as torch bug fix
e066869
raw
history blame
226 Bytes
gradio
torch==1.13.1 --index-url https://download.pytorch.org/whl/cu118
torchvision==0.14.1 --index-url https://download.pytorch.org/whl/cu118
diffusers
transformers
accelerate
scipy
git+https://github.com/davidbau/baukit.git