jax[cuda11_cudnn82]
jaxlib
flax
transformers
git+https://github.com/huggingface/diffusers@main