File size: 714 Bytes
fab4085
 
 
 
ae925e4
fab4085
756dcdd
fab4085
 
 
756dcdd
 
fab4085
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
756dcdd
42bcfa6
 
fab4085
756dcdd
65579be
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
-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