Contrebande Labs commited on
Commit
236f801
·
1 Parent(s): 2b3e373

use space-local CUDA

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_pip]
3
  flax
4
  optax
5
  diffusers
 
1
  -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
2
+ jax[cuda11_local]
3
  flax
4
  optax
5
  diffusers