diff --git a/CMakeLists.txt b/CMakeLists.txt index 513f4a87f..edc64f877 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -45,7 +45,7 @@ set(HIP_SUPPORTED_ARCHS "gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1 # requirements.txt files and should be kept consistent. The ROCm torch # versions are derived from docker/Dockerfile.rocm # -set(TORCH_SUPPORTED_VERSION_CUDA "2.7.0") +set(TORCH_SUPPORTED_VERSION_CUDA "2.7.1") set(TORCH_SUPPORTED_VERSION_ROCM "2.7.0") # diff --git a/pyproject.toml b/pyproject.toml index 65ba0b4d8..85a112ff5 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -6,7 +6,7 @@ requires = [ "packaging>=24.2", "setuptools>=77.0.3,<80.0.0", "setuptools-scm>=8.0", - "torch == 2.7.0", + "torch == 2.7.1", "wheel", "jinja2", ] diff --git a/requirements/build.txt b/requirements/build.txt index 528cd3b53..dd644d621 100644 --- a/requirements/build.txt +++ b/requirements/build.txt @@ -4,7 +4,7 @@ ninja packaging>=24.2 setuptools>=77.0.3,<80.0.0 setuptools-scm>=8 -torch==2.7.0 +torch==2.7.1 wheel jinja2>=3.1.6 regex diff --git a/requirements/cuda.txt b/requirements/cuda.txt index a71d9728f..c1273b224 100644 --- a/requirements/cuda.txt +++ b/requirements/cuda.txt @@ -6,9 +6,9 @@ numba == 0.61.2; python_version > '3.9' # Dependencies for NVIDIA GPUs ray[cgraph]>=2.43.0, !=2.44.* # Ray Compiled Graph, required for pipeline parallelism in V1. -torch==2.7.0 -torchaudio==2.7.0 +torch==2.7.1 +torchaudio==2.7.1 # These must be updated alongside torch -torchvision==0.22.0 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version -# https://github.com/facebookresearch/xformers/releases/tag/v0.0.30 -xformers==0.0.30; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch >= 2.7 +torchvision==0.22.1 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version +# https://github.com/facebookresearch/xformers/releases/tag/v0.0.31 +xformers==0.0.31; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch >= 2.7 diff --git a/requirements/test.in b/requirements/test.in index e8537d10f..e8715afaf 100644 --- a/requirements/test.in +++ b/requirements/test.in @@ -22,9 +22,9 @@ sentence-transformers # required for embedding tests soundfile # required for audio tests jiwer # required for audio tests timm # required for internvl test -torch==2.7.0 -torchaudio==2.7.0 -torchvision==0.22.0 +torch==2.7.1 +torchaudio==2.7.1 +torchvision==0.22.1 transformers_stream_generator # required for qwen-vl test mamba_ssm # required for plamo2 test matplotlib # required for qwen-vl test diff --git a/requirements/test.txt b/requirements/test.txt index 84303b831..90d8f8ff0 100644 --- a/requirements/test.txt +++ b/requirements/test.txt @@ -762,7 +762,7 @@ tomli==2.2.1 # via schemathesis tomli-w==1.2.0 # via schemathesis -torch==2.7.0+cu128 +torch==2.7.1+cu128 # via # -r requirements/test.in # accelerate @@ -781,12 +781,12 @@ torch==2.7.0+cu128 # torchvision # vector-quantize-pytorch # vocos -torchaudio==2.7.0+cu128 +torchaudio==2.7.1+cu128 # via # -r requirements/test.in # encodec # vocos -torchvision==0.22.0+cu128 +torchvision==0.22.1+cu128 # via # -r requirements/test.in # timm @@ -816,7 +816,7 @@ transformers==4.53.2 # transformers-stream-generator transformers-stream-generator==0.0.5 # via -r requirements/test.in -triton==3.3.0 +triton==3.3.1 # via torch tritonclient==2.51.0 # via diff --git a/tests/entrypoints/openai/test_vision.py b/tests/entrypoints/openai/test_vision.py index fd613842f..b6f1d6480 100644 --- a/tests/entrypoints/openai/test_vision.py +++ b/tests/entrypoints/openai/test_vision.py @@ -36,11 +36,11 @@ EXPECTED_MM_BEAM_SEARCH_RES = [ ], [ "The image shows a Venn diagram with three over", - "This image shows a Venn diagram with three over", + "The image shows a Venn diagram with three intersect", ], [ "This image displays a gradient of colors ranging from", - "This image displays a gradient of colors transitioning from", + "The image displays a gradient of colors ranging from", ], ]