diff --git a/pyproject.toml b/pyproject.toml index 51a53e21ed..5ad5101c01 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -166,7 +166,7 @@ pin_pytorch_cpu = [ "torch; platform_machine=='x86_64' and platform_system == 'Darwin'", ] pin_pytorch_gpu = [ - "torch>=2.7,<2.9", + "torch>=2.7,<2.10", ] pin_jax = [ "jax==0.5.0;python_version>='3.10'",