bguisard commited on
Commit
e330f84
·
1 Parent(s): 2ee086f

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- jax
2
  flax
3
  transformers
4
  git+https://github.com/huggingface/diffusers
 
1
+ jax[cuda11_cudnn82]
2
  flax
3
  transformers
4
  git+https://github.com/huggingface/diffusers