p

Signed-off-by: khluu <khluu000@gmail.com>
2026-04-02 11:47:14 -07:00
parent 08ed2b9688
commit 0803bf18fd
2 changed files with 5 additions and 5 deletions
@@ -330,7 +330,8 @@ RUN --mount=type=bind,from=export_vllm,src=/,target=/install \
    && uv pip install --system -r requirements/rocm.txt \
    && uv pip install --system -r requirements/rocm-test.txt \
    && pip uninstall -y vllm \
-    && uv pip install --system *.whl
+    && uv pip install --system *.whl \
+    && uv pip install --system "transformers==5.5.0"

 # Verify that PyTorch is the ROCm build, not CUDA
 RUN python3 -c "import torch; assert torch.version.hip is not None, \
@@ -428,6 +429,7 @@ RUN --mount=type=bind,from=export_vllm,src=/,target=/install \
    && uv pip install --system -r requirements/rocm.txt \
    && pip uninstall -y vllm \
    && uv pip install --system *.whl
+    && uv pip install --system "transformers==5.5.0"

 ARG COMMON_WORKDIR
 ARG BASE_IMAGE