tsungtao commited on
Commit
7585942
1 Parent(s): d58e7b0

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,5 +1,5 @@
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
 
1
  -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
2
+ jax[cuda11_cudnn82]
3
  jaxlib
4
  git+https://github.com/huggingface/diffusers@main
5
  opencv-python