bguisard commited on
Commit
edf436d
1 Parent(s): b6c423f

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- jax[cuda11_cudnn82]
2
  jaxlib
3
  flax
4
  transformers
 
1
+ jax[cuda]
2
  jaxlib
3
  flax
4
  transformers