transformers flax --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html jaxlib git+https://github.com/huggingface/diffusers@main pillow opencv-python torch torchvision accelerate gradio --upgrade