xformers git+https://github.com/huggingface/diffusers@main -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html jax[cuda11_pip] jaxlib opencv-python transformers flax pillow -f https://download.pytorch.org/whl/torch_stable.html torch==1.12.0+cu113 torchvision==0.13.0+cu113