File size: 239 Bytes
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_cudnn805] #jax[cuda11_cudnn86] #jax[cuda11_cudnn805]
flax