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