alexlau commited on
Commit
b04a705
1 Parent(s): eaa31f6

downgrade flax to 0.3.4

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -3,7 +3,7 @@ jax[cuda111]
3
  numpy>=1.16.4
4
  jax==0.2.17
5
  jaxlib==0.1.69
6
- flax==0.3.6
7
  # opencv-python>=4.4.0
8
  Pillow>=7.2.0
9
  pyyaml>=5.3.1
 
3
  numpy>=1.16.4
4
  jax==0.2.17
5
  jaxlib==0.1.69
6
+ flax==0.3.4
7
  # opencv-python>=4.4.0
8
  Pillow>=7.2.0
9
  pyyaml>=5.3.1