juancopi81 commited on
Commit
8f8dcb6
1 Parent(s): fe63963

Change to jax cpu

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
  tensorflow
2
  nest-asyncio
3
  pyfluidsynth==1.3.0
4
- jax[cuda]==0.3.15 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
5
  # pin CLU for python 3.7 compatibility
6
  clu==0.0.7
7
  # pin Orbax to use Checkpointer
 
1
  tensorflow
2
  nest-asyncio
3
  pyfluidsynth==1.3.0
4
+ jax[cpu]
5
  # pin CLU for python 3.7 compatibility
6
  clu==0.0.7
7
  # pin Orbax to use Checkpointer