8c4daf1 5b09d17 8c4daf1
1
2
3
4
5
6
7
8
9
10
11
numpy pillow transformers diffusers einops -f https://download.pytorch.org/whl/cpu/torch torch[cpu] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html jax[cuda11_pip] #jax[cuda11_cudnn82] #jax[cuda11_cudnn86] #jax[cuda11_cudnn805] flax
