merve HF staff commited on
Commit
85d051b
1 Parent(s): 2ba1189

Update requirements.txt

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