mfidabel commited on
Commit
d07d7a8
1 Parent(s): de3193d

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -1,8 +1,8 @@
 
 
 
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
- numpy
7
- transformers
8
- flax
 
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