From 1680dacfee42aaa2709e594e7fb4ae4f09741ad9 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 7 May 2026 02:45:13 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- studio/install_python_stack.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/studio/install_python_stack.py b/studio/install_python_stack.py index 9251827b50..c6bbfda385 100644 --- a/studio/install_python_stack.py +++ b/studio/install_python_stack.py @@ -41,8 +41,8 @@ IS_MAC_INTEL = IS_MACOS and platform.machine() == "x86_64" # Mapping from detected ROCm (major, minor) to the best PyTorch wheel tag on # download.pytorch.org. Entries are checked newest-first (>=). _ROCM_TORCH_INDEX: dict[tuple[int, int], str] = { - (7, 2): "rocm7.2", # torch 2.11.0 - (7, 1): "rocm7.1", # torch 2.10.0 + (7, 2): "rocm7.2", # torch 2.11.0 + (7, 1): "rocm7.1", # torch 2.10.0 (7, 0): "rocm7.0", (6, 4): "rocm6.4", (6, 3): "rocm6.3",