Hugging Face's logo

Spaces:
flax-community
/
DietNerf-Demo
Runtime error

App Files Community
Alex Lau commited on
Commit
3745caa
•
1 Parent(s): 48b917c

change jax release url link

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  # in CUDA mode, can set src.config.py::DOWNSAMPLE=2
2
- --find-links https://storage.googleapis.com/jax-releases/jax_releases.html
3
  jax[cuda112]==0.2.17
4
  # in CPU mode, set src.config.py::DOWNSAMPLE=4
5
  #jax[cpu]==0.2.17
 
1
  # in CUDA mode, can set src.config.py::DOWNSAMPLE=2
2
+ --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
3
  jax[cuda112]==0.2.17
4
  # in CPU mode, set src.config.py::DOWNSAMPLE=4
5
  #jax[cpu]==0.2.17