correct orbax dependency

LeonGerBerlin · web-flow · commit a2d7137eb80f · 2023-10-16T09:46:05.000+02:00
diff --git a/setup.cfg b/setup.cfg
@@ -25,6 +25,7 @@ install_requires =
     jax==0.3.23
     jaxlib==0.3.22
     orbax-checkpoint==0.1.1
+    orbax==0.1.2
     dm-haiku
     optax
     chex==0.1.5
