Update generate_binary_build_matrix.py

pytorch · Nov 26, 2024 · e1c8585 · e1c8585
1 parent 04a487f
commit e1c8585
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -148,9 +148,9 @@ def initialize_globals(channel: str, build_python_only: bool) -> None:
     else:
         PYTHON_ARCHES = PYTHON_ARCHES_DICT[channel]
     WHEEL_CONTAINER_IMAGES = {
-       "11.8": f"pytorch/manylinux-builder:cuda11.8",
-       "12.4": f"pytorch/manylinux-builder:cuda12.4",
-       "12.6": f"pytorch/manylinux2_28-builder:cuda12.6",
+       "11.8": "pytorch/manylinux-builder:cuda11.8",
+       "12.4": "pytorch/manylinux-builder:cuda12.4",
+       "12.6": "pytorch/manylinux2_28-builder:cuda12.6",
         **{
             gpu_arch: f"pytorch/manylinux-builder:rocm{gpu_arch}"
             for gpu_arch in ROCM_ARCHES