From 6315126bd1347f6de43955c789697081a09384e9 Mon Sep 17 00:00:00 2001
From: Juan Acevedo <juancevedo@gmail.com>
Date: Mon, 9 Dec 2024 19:34:20 +0000
Subject: [PATCH] updates flax dependency to remove error with jax >=0.4.36

---
 requirements.txt | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 5202dc2..82ade3e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,7 +4,7 @@ grain-nightly
 google-cloud-storage==2.17.0
 absl-py
 datasets
-flax==0.8.5
+flax>=0.10.2
 optax>=0.2.3
 torch>=2.3.1
 torchvision>=0.18.1
@@ -26,6 +26,4 @@ git+https://github.com/mlperf/logging.git
 opencv-python-headless==4.10.0.84
 orbax-checkpoint>=0.5.20
 tokenizers==0.20.0
-huggingface_hub==0.24.7
-
 huggingface_hub==0.24.7
\ No newline at end of file