From 757113113c105456037817474097ba852835f711 Mon Sep 17 00:00:00 2001 From: Mayank Mishra <32954280+mayank31398@users.noreply.github.com> Date: Thu, 16 Apr 2026 19:48:54 -0700 Subject: [PATCH] Revert "[UV] update torch version (#414)" This reverts commit 5906d3adcf03bce19d4d492d0cfc93c7e862ae17. --- pyproject.toml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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", ]