revert to CPU setting
368e60c
Alex Lau
commited on
Spaces:
Build error
Build error
Commit History
change jax release url link
3745caa
Alex Lau
commited on
revert to orig jax dependencies
48b917c
cpu jax installation
6d2eea0
change jax cuda version
2244280
revert to cpu mode with downsample=4
61a957f
set CUDA jax
279af7e
downgrade jaxlib
0c4c744
spec jax cpu install
96899f1
comment jax cuda
6dba29d
downgrade flax to 0.3.4
b04a705
revert to CPU and fix jax, jaxlib, flax ver
eaa31f6
Revert "revert to cpu config"
bf2d3d5
revert to gpu mode
40a686d
revert to cpu config
b30f1b8
Add JAX CUDA version in requirements.txt
fc5bccc
Fixed requirements.txt
4caa204
Sasikanth
commited on