Skip to content

Commit d7f15ad

Browse files
Alexei-V-Ivanov-AMDtjtanaa
authored andcommitted
[Bugfix] Fixing AMD LoRA CI test. (vllm-project#12329)
Signed-off-by: Alexei V. Ivanov <[email protected]>
1 parent 7b79dad commit d7f15ad

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

Dockerfile.rocm

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,8 @@ COPY --from=build_vllm ${COMMON_WORKDIR}/vllm /vllm-workspace
7272
RUN cd /vllm-workspace \
7373
&& rm -rf vllm \
7474
&& python3 -m pip install -e tests/vllm_test_utils \
75-
&& python3 -m pip install lm-eval[api]==0.4.4
75+
&& python3 -m pip install lm-eval[api]==0.4.4 \
76+
&& python3 -m pip install pytest-shard
7677

7778
# -----------------------
7879
# Final vLLM image

0 commit comments

Comments
 (0)