diff --git a/pyproject.toml b/pyproject.toml index 1c03e9e17be5..49daf0a2495d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -6,7 +6,7 @@ requires = [ "packaging", "setuptools>=61", "setuptools-scm>=8.0", - "torch == 2.5.1", + "torch == 2.8.0.dev20250419", "wheel", "jinja2", ] diff --git a/requirements-build.txt b/requirements-build.txt index fec01caaf25e..35600e14368f 100644 --- a/requirements-build.txt +++ b/requirements-build.txt @@ -4,6 +4,6 @@ ninja packaging setuptools>=61 setuptools-scm>=8 -torch==2.5.1 +torch==2.8.0.dev20250419 wheel jinja2 diff --git a/requirements-cpu.txt b/requirements-cpu.txt index 9491e27d1270..9d08054a2a19 100644 --- a/requirements-cpu.txt +++ b/requirements-cpu.txt @@ -4,7 +4,7 @@ # Dependencies for CPUs torch==2.5.1+cpu; platform_machine != "ppc64le" and platform_machine != "aarch64" and platform_system != "Darwin" and platform_machine != "s390x" torch==2.5.1; platform_machine == "ppc64le" or platform_machine == "aarch64" or platform_system == "Darwin" -torch==2.7.0.dev20250304; platform_machine == "s390x" +torch==2.8.0.dev20250419; platform_machine == "s390x" # required for the image processor of minicpm-o-2_6, this must be updated alongside torch torchaudio; platform_machine != "ppc64le" and platform_machine != "s390x" diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 55c47eb48177..36c4c9cffa64 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -5,7 +5,7 @@ numba == 0.60.0 # v0.61 doesn't support Python 3.9. Required for N-gram speculat # Dependencies for NVIDIA GPUs ray[adag] == 2.40.0 # Required for pipeline parallelism in V1. -torch == 2.5.1 +torch == 2.8.0.dev20250419 torchaudio==2.5.1 # These must be updated alongside torch torchvision == 0.20.1 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version diff --git a/requirements-openvino.txt b/requirements-openvino.txt index ac9d851d661b..d7b942b08057 100644 --- a/requirements-openvino.txt +++ b/requirements-openvino.txt @@ -1,7 +1,7 @@ # Common dependencies -r requirements-common.txt -torch == 2.5.1 # should be aligned with "common" vLLM torch version +torch == 2.8.0.dev20250419 # 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 # latest optimum is used to support latest transformers version diff --git a/requirements-rocm-build.txt b/requirements-rocm-build.txt index 00ae0340fc52..b95475058173 100644 --- a/requirements-rocm-build.txt +++ b/requirements-rocm-build.txt @@ -2,7 +2,7 @@ -r requirements-common.txt --extra-index-url https://download.pytorch.org/whl/rocm6.2 -torch==2.5.1 +torch==2.8.0.dev20250419 torchvision==0.20.1 torchaudio==2.5.1 diff --git a/requirements-test.in b/requirements-test.in index 53c531360d87..19785e30e8ab 100644 --- a/requirements-test.in +++ b/requirements-test.in @@ -21,7 +21,7 @@ sentence-transformers # required for embedding tests soundfile # required for audio tests jiwer # required for audio tests timm # required for internvl test -torch==2.5.1 +torch==2.8.0.dev20250419 torchaudio==2.5.1 transformers_stream_generator # required for qwen-vl test matplotlib # required for qwen-vl test diff --git a/requirements-tpu.txt b/requirements-tpu.txt index 1abde714af7c..84fb05670e43 100644 --- a/requirements-tpu.txt +++ b/requirements-tpu.txt @@ -17,7 +17,7 @@ ray[default] --find-links https://storage.googleapis.com/libtpu-releases/index.html --find-links https://storage.googleapis.com/jax-releases/jax_nightly_releases.html --find-links https://storage.googleapis.com/jax-releases/jaxlib_nightly_releases.html -torch==2.6.0.dev20241216+cpu +torch==2.8.0.dev20250419 torch_xla[tpu, pallas] @ https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.7.0.dev20250124-cp39-cp39-linux_x86_64.whl ; python_version == "3.9" torch_xla[tpu, pallas] @ https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.7.0.dev20250124-cp310-cp310-linux_x86_64.whl ; python_version == "3.10" torch_xla[tpu, pallas] @ https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.7.0.dev20250124-cp311-cp311-linux_x86_64.whl ; python_version == "3.11"