requirements.txt

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