Erasing-Concepts-In-Diffusion / requirements.txt
Damian Stewart
wip adding AMP and xformers to training code path
fc73e59
raw
history blame
252 Bytes
gradio
torch --index-url https://download.pytorch.org/whl/cu118
torchvision --index-url https://download.pytorch.org/whl/cu118
diffusers
transformers
accelerate
scipy
git+https://github.com/davidbau/baukit.git
xformers
bitsandbytes==0.38.1
safetensors