Upgrade orbax

apple · Jan 30, 2025 · 140a18f · 140a18f
1 parent c1a476d
commit 140a18f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -150,7 +150,7 @@ mmau = [
 # Orbax checkpointing.
 orbax = [
     "humanize==4.10.0",
-    "orbax-checkpoint==0.6.4",
+    "orbax-checkpoint==0.11.1",
 ]
 # Grain input processing. Currently does not support macos.
 grain = [