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

Signed-off-by: Alexei V. Ivanov <[email protected]>
vllm-project · Jan 23, 2025 · 8d7aa9d · 8d7aa9d
1 parent 68c4421
commit 8d7aa9d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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