revert to torch 2.5.1 due to compatibility with torchvision.

jfacevedo-google · jfacevedo-google · commit 4f230da1c193 · 2025-04-22T18:34:47.000Z
diff --git a/requirements.txt b/requirements.txt
@@ -6,7 +6,7 @@ absl-py
 datasets
 flax>=0.10.2
 optax>=0.2.3
-torch==2.6.0
+torch==2.5.1
 torchvision==0.20.1
 ftfy
 tensorboard>=2.17.0
diff --git a/requirements_with_jax_stable_stack.txt b/requirements_with_jax_stable_stack.txt
@@ -29,6 +29,6 @@ tensorboardx==2.6.2.2
 tensorflow>=2.17.0
 tensorflow-datasets>=4.9.6
 tokenizers==0.21.0
-torch==2.6.0
+torch==2.5.1
 torchvision==0.20.1
 transformers==4.48.1
