Vincent-luo commited on
Commit
9c5ac71
1 Parent(s): 02bfba0

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,10 +1,10 @@
 
 
 
1
  -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
2
- jax[cuda11_cudnn805]
3
  jaxlib
4
  git+https://github.com/huggingface/diffusers@main
5
  opencv-python
6
- transformers
7
- flax
8
  torch
9
  torchvision
10
  mediapipe==0.9.1
 
1
+ transformers
2
+ flax
3
+ jax[cuda11_pip]
4
  -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 
5
  jaxlib
6
  git+https://github.com/huggingface/diffusers@main
7
  opencv-python
 
 
8
  torch
9
  torchvision
10
  mediapipe==0.9.1