pin pygrain to 0.0.10

coolkp · coolkp · commit c26ea5a7a49c · 2025-03-21T03:29:00.000Z
Signed-off-by: kunjan &lt;kunjanp@google.com&gt;
diff --git 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
