From a04d8372e4ad743417b37a16e22f3bdefae66abd Mon Sep 17 00:00:00 2001
From: Juan Acevedo <juancevedo@gmail.com>
Date: Fri, 20 Dec 2024 11:50:19 -0800
Subject: [PATCH] update orbax dependency due to breaking changes in 0.10.3
 (#139)

---
 requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 82ade3e..2c7b4bd 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -24,6 +24,6 @@ tensorflow-datasets>=4.9.6
 ruff>=0.1.5,<=0.2
 git+https://github.com/mlperf/logging.git
 opencv-python-headless==4.10.0.84
-orbax-checkpoint>=0.5.20
+orbax-checkpoint==0.10.2
 tokenizers==0.20.0
 huggingface_hub==0.24.7
\ No newline at end of file