Merge branch 'main' into multi_res_support

jfacevedo-google · jfacevedo-google · commit a53b9bcafab8 · 2025-04-22T18:31:55.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.5.1
+torch==2.6.0
 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.5.1
+torch==2.6.0
 torchvision==0.20.1
 transformers==4.48.1
