[AMD] Pin compressed-tensors<0.16.0 for srt_hip (fixes ROCm 7.2 nightly build) (#26591)

This commit is contained in:
YC Yen-Ching Tseng
2026-05-29 11:34:45 +08:00
committed by GitHub
parent b1173c8c14
commit 272066566f
+7
View File
@@ -95,6 +95,13 @@ srt_hip = [
"torch",
"petit_kernel==0.0.2",
"wave-lang==3.8.2",
# HOTFIX (2026-05-28): compressed-tensors 0.16.0 added `torch>=2.10.0`,
# which forces pip off the ROCm wheel (torch 2.9.1+rocm7.2) and silently
# swaps it for the PyPI default `torch 2.12.0+cu130`, breaking the
# downstream HIP build of fast-hadamard-transform with a NameError on
# `bare_metal_version`. Pin below 0.16.0 until the ROCm base ships
# torch>=2.10.
"compressed-tensors<0.16.0",
]
diffusion_hip = [