From fccb25a33539c3f3488b4603506c0c7aa52511be Mon Sep 17 00:00:00 2001 From: Param Bole Date: Fri, 25 Apr 2025 23:13:48 +0000 Subject: [PATCH 1/2] Fix 413751357 updating orbax version which is lowering the JAX version --- requirements.txt | 2 +- requirements_with_jax_stable_stack.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index 62a3819e..fc4514f5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -24,7 +24,7 @@ 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.10.3 +orbax-checkpoint>=0.10.3 tokenizers==0.21.0 huggingface_hub==0.30.2 transformers==4.48.1 diff --git a/requirements_with_jax_stable_stack.txt b/requirements_with_jax_stable_stack.txt index 80ad1434..fd3ded3d 100644 --- a/requirements_with_jax_stable_stack.txt +++ b/requirements_with_jax_stable_stack.txt @@ -14,7 +14,7 @@ jaxlib>=0.4.30 Jinja2 opencv-python-headless==4.10.0.84 optax>=0.2.3 -orbax-checkpoint==0.10.3 +orbax-checkpoint>=0.10.3 parameterized Pillow pyink From 3831423e1b1441548e8ea88a06d21642062498b5 Mon Sep 17 00:00:00 2001 From: Param Bole Date: Fri, 25 Apr 2025 23:40:41 +0000 Subject: [PATCH 2/2] Updating the right dependency --- requirements.txt | 4 ++-- requirements_with_jax_stable_stack.txt | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index fc4514f5..defbb151 100644 --- a/requirements.txt +++ b/requirements.txt @@ -19,12 +19,12 @@ Pillow pylint pyink pytest==8.2.2 -tensorflow==2.17.0 +tensorflow>=2.17.0 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.10.3 +orbax-checkpoint==0.10.3 tokenizers==0.21.0 huggingface_hub==0.30.2 transformers==4.48.1 diff --git a/requirements_with_jax_stable_stack.txt b/requirements_with_jax_stable_stack.txt index fd3ded3d..80ad1434 100644 --- a/requirements_with_jax_stable_stack.txt +++ b/requirements_with_jax_stable_stack.txt @@ -14,7 +14,7 @@ jaxlib>=0.4.30 Jinja2 opencv-python-headless==4.10.0.84 optax>=0.2.3 -orbax-checkpoint>=0.10.3 +orbax-checkpoint==0.10.3 parameterized Pillow pyink