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