diff --git a/pyproject.toml b/pyproject.toml index ac86a9e45..77813e63a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -31,7 +31,7 @@ Repository = "https://github.com/open-lm-engine/lm-engine" [project.optional-dependencies] cuda = [ - "torch>=2.11.0", + "torch>=2.10.0", ] dev = [ "pre-commit>=4.4.0", @@ -41,12 +41,12 @@ mamba2 = [ "mamba-ssm>=2.2.6.post3", ] tpu = [ - "torch>=2.11.0", + "torch==2.10.0", "torch-xla[pallas,tpu]==2.9.0", ] data = [ "ray[default]>=2.52.1", - "torch>=2.11.0", + "torch>=2.10.0", "zstandard>=0.25.0", "aiobotocore>=3.0.0", ]