diff --git a/requirements/build.txt b/requirements/build.txt index 880130beb12..5012276fa01 100644 --- a/requirements/build.txt +++ b/requirements/build.txt @@ -4,6 +4,6 @@ ninja packaging>=24.2 setuptools>=77.0.3,<81.0.0 setuptools_scm>=8 -torch==2.7.0 # Corresponds to the version used by vLLM main branch +torch==2.12.0 # Corresponds to the version used by vLLM main branch wheel xxhash==3.5.0 diff --git a/requirements/common.txt b/requirements/common.txt index 34e30c28d07..042d016a534 100644 --- a/requirements/common.txt +++ b/requirements/common.txt @@ -15,6 +15,6 @@ safetensors setuptools>=77.0.3,<81.0.0 setuptools_scm>=8 sortedcontainers -torch==2.7.0 # Should correspond to the version used by vLLM main branch +torch==2.12.0 # Should correspond to the version used by vLLM main branch transformers >= 4.51.1 xxhash==3.5.0 diff --git a/requirements/cuda.txt b/requirements/cuda.txt index 719261d725f..f762945eda9 100644 --- a/requirements/cuda.txt +++ b/requirements/cuda.txt @@ -6,6 +6,6 @@ ray >= 2.9 nvidia-ml-py # for pynvml package # These must be updated alongside torch to correspond to vLLM versions -torch == 2.7.0 +torch == 2.12.0 torchvision == 0.22.0 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version xformers == 0.0.30; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch 2.7.0