diff --git a/requirements.txt b/requirements.txt index e26b45b8..ce04e1b0 100644 --- 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.7.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 index 5a88c800..f1046c85 100644 --- a/requirements_with_jax_stable_stack.txt +++ b/requirements_with_jax_stable_stack.txt @@ -29,7 +29,7 @@ tensorboardx==2.6.2.2 tensorflow>=2.17.0 tensorflow-datasets>=4.9.6 tokenizers==0.21.0 -torch==2.5.1 +torch==2.7.0 torchvision==0.20.1 transformers==4.48.1 imageio==2.37.0