diff --git a/CMakeLists.txt b/CMakeLists.txt index 1a6a311e97633..943424bc4edfa 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -49,8 +49,8 @@ set(HIP_SUPPORTED_ARCHS "gfx906;gfx908;gfx90a;gfx940;gfx941;gfx942;gfx1030;gfx11 # requirements.txt files and should be kept consistent. The ROCm torch # versions are derived from Dockerfile.rocm # -set(TORCH_SUPPORTED_VERSION_CUDA "2.5.0") -set(TORCH_SUPPORTED_VERSION_ROCM "2.5.0") +set(TORCH_SUPPORTED_VERSION_CUDA "2.5.1") +set(TORCH_SUPPORTED_VERSION_ROCM "2.5.1") # # Try to find python package with an executable that exactly matches diff --git a/pyproject.toml b/pyproject.toml index e78f5652f486b..0bbab3cd3fbc3 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -6,7 +6,7 @@ requires = [ "packaging", "setuptools>=61", "setuptools-scm>=8.0", - "torch == 2.5.0", + "torch == 2.5.1", "wheel", "jinja2", ] diff --git a/requirements-build.txt b/requirements-build.txt index 7b16d9778c1a6..fec01caaf25ef 100644 --- a/requirements-build.txt +++ b/requirements-build.txt @@ -4,6 +4,6 @@ ninja packaging setuptools>=61 setuptools-scm>=8 -torch==2.5.0 +torch==2.5.1 wheel jinja2 diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 282ab11838bf4..058ab7c1ee9df 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -4,7 +4,7 @@ # Dependencies for NVIDIA GPUs ray >= 2.9 nvidia-ml-py >= 12.560.30 # for pynvml package -torch == 2.5.0 +torch == 2.5.1 # These must be updated alongside torch -torchvision == 0.20 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version -xformers == 0.0.28.post2; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch 2.5.0 +torchvision == 0.20.1 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version +xformers == 0.0.28.post3; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch 2.5.1 diff --git a/requirements-openvino.txt b/requirements-openvino.txt index 7ad0d1e7f704b..95e5914757812 100644 --- a/requirements-openvino.txt +++ b/requirements-openvino.txt @@ -1,7 +1,7 @@ # Common dependencies -r requirements-common.txt -torch == 2.5.0 # should be aligned with "common" vLLM torch version +torch == 2.5.1 # should be aligned with "common" vLLM torch version openvino >= 2024.4.0 # since 2024.4.0 both CPU and GPU support Paged Attention optimum @ git+https://github.com/huggingface/optimum.git@main # latest optimum is used to support latest transformers version diff --git a/requirements-test.in b/requirements-test.in index 5d44664c082a6..560c005fd6157 100644 --- a/requirements-test.in +++ b/requirements-test.in @@ -18,7 +18,7 @@ ray[adag]==2.35 sentence-transformers # required for embedding soundfile # required for audio test timm # required for internvl test -torch==2.5.0 +torch==2.5.1 transformers_stream_generator # required for qwen-vl test matplotlib # required for qwen-vl test datamodel_code_generator # required for minicpm3 test diff --git a/requirements-test.txt b/requirements-test.txt index 7477b7c3a79cd..518e81021cbcb 100644 --- a/requirements-test.txt +++ b/requirements-test.txt @@ -492,7 +492,7 @@ timm==1.0.11 # via -r requirements-test.in tokenizers==0.20.1 # via transformers -torch==2.5.0 +torch==2.5.1 # via # -r requirements-test.in # accelerate @@ -503,7 +503,7 @@ torch==2.5.0 # tensorizer # timm # torchvision -torchvision==0.20.0 +torchvision==0.20.1 # via timm tqdm==4.66.6 # via