Cherry pick [ROCm] [CI] [Release] Rocm wheel pipeline with sccache #32264

Signed-off-by: Kevin H. Luu <khluu000@gmail.com>
2026-01-16 02:56:18 +08:00
parent 0e31fc7996
commit c2a37a3cf8
10 changed files with 1343 additions and 25 deletions
--- a/requirements/rocm-test.txt
+++ b/requirements/rocm-test.txt
@@ -80,6 +80,8 @@ num2words==0.5.14
 pqdm==0.2.0
    # via lm-eval

+# Required for fastsafetensors test
+fastsafetensors @ git+https://github.com/foundation-model-stack/fastsafetensors.git@d6f998a03432b2452f8de2bb5cefb5af9795d459
 # Required for suffix decoding test
 arctic-inference == 0.1.1
 # Required for Nemotron test
--- a/requirements/rocm.txt
+++ b/requirements/rocm.txt
@@ -15,5 +15,4 @@ setuptools-scm>=8
 runai-model-streamer[s3,gcs]==0.15.3
 conch-triton-kernels==1.2.1
 timm>=1.0.17
-fastsafetensors @ git+https://github.com/foundation-model-stack/fastsafetensors.git@d6f998a03432b2452f8de2bb5cefb5af9795d459
 grpcio-tools>=1.76.0