juancopi81 commited on
Commit
fe63963
1 Parent(s): 6332e4c

Different jax package

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -2
requirements.txt CHANGED
@@ -1,5 +1,8 @@
1
  tensorflow
2
- jaxlib==0.3.15+cuda11.cudnn82
3
  nest-asyncio
4
  pyfluidsynth==1.3.0
5
- jax[cuda]==0.3.15 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 
 
 
 
 
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
8
+ orbax==0.0.2