transformers flax jax[cuda11_pip] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html jaxlib git+https://github.com/huggingface/diffusers@main opencv-python --extra-index-url https://download.pytorch.org/whl/cu113 torch torchvision git+https://github.com/facebookresearch/segment-anything.git accelerate