| -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html | |
| ConfigArgParse==1.7 | |
| Pillow==10.0.0 | |
| accelerate | |
| chex==0.1.7 | |
| diffusers | |
| einops==0.6.1 | |
| flax==0.6.10 | |
| flaxmodels==0.1.3 | |
| gradio | |
| huggingface_hub | |
| jax==0.4.11 | |
| jaxlib==0.4.11+cuda11.cudnn86 | |
| jaxtyping==0.2.20 | |
| ml-dtypes==0.1.0 | |
| numpy==1.24.1 | |
| nvidia-cublas-cu11==11.11.3.6 | |
| nvidia-cuda-cupti-cu11==11.8.87 | |
| nvidia-cuda-nvcc-cu11==11.8.89 | |
| nvidia-cuda-runtime-cu11==11.8.89 | |
| nvidia-cudnn-cu11==8.9.2.26 | |
| nvidia-cufft-cu11==10.9.0.58 | |
| nvidia-cusolver-cu11==11.4.1.48 | |
| nvidia-cusparse-cu11==11.7.5.86 | |
| opt-einsum==3.3.0 | |
| optax==0.2.0 | |
| orbax-checkpoint==0.2.4 | |
| peft==0.6.2 | |
| pillow==10.0.0 | |
| scipy==1.10.1 | |
| timm | |
| torch | |
| torchvision | |
| tqdm==4.65.0 | |
| transformers | |
| wandb | |
| xformers==0.0.23 | |