From c26ea5a7a49c190cb7cc4f10ef4260348aeed5c0 Mon Sep 17 00:00:00 2001 From: kunjan Date: Fri, 21 Mar 2025 03:29:00 +0000 Subject: [PATCH] pin pygrain to 0.0.10 Signed-off-by: kunjan --- requirements_with_jax_stable_stack.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements_with_jax_stable_stack.txt b/requirements_with_jax_stable_stack.txt index 869d2d88c..ba21b4774 100644 --- a/requirements_with_jax_stable_stack.txt +++ b/requirements_with_jax_stable_stack.txt @@ -7,7 +7,7 @@ flax>=0.10.2 ftfy git+https://github.com/mlperf/logging.git google-cloud-storage==2.17.0 -grain-nightly +grain-nightly==0.0.10 huggingface_hub==0.24.7 jax>=0.4.30 jaxlib>=0.4.30