Erasing-Concepts-In-Diffusion / requirements.txt
Damian Stewart
add tensorboard, validation, sample output
50b9662
raw
history blame contribute delete
264 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
tensorboard