[Bugfix] Fixing AMD LoRA CI test. (#12329)

Signed-off-by: Alexei V. Ivanov <alexei.ivanov@amd.com>

[Bugfix] Fixing AMD LoRA CI test. (#12329)
Signed-off-by: Alexei V. Ivanov <alexei.ivanov@amd.com>
8d7aa9de · Alexei-V-Ivanov-AMD · GitHub · 68c4421b · 8d7aa9de
Unverified Commit 8d7aa9de authored Jan 22, 2025 by Alexei-V-Ivanov-AMD Committed by GitHub Jan 23, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

Dockerfile.rocm Dockerfile.rocm +2 -1

No files found.
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -72,7 +72,8 @@ COPY --from=build_vllm ${COMMON_WORKDIR}/vllm /vllm-workspace
 RUN cd /vllm-workspace \
    && rm -rf vllm \
    && python3 -m pip install -e tests/vllm_test_utils \
-    && python3 -m pip install lm-eval[api]==0.4.4
+    && python3 -m pip install lm-eval[api]==0.4.4 \
+    && python3 -m pip install pytest-shard

 # -----------------------
 # Final vLLM image