We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 68c4421 commit 8d7aa9dCopy full SHA for 8d7aa9d
Dockerfile.rocm
@@ -72,7 +72,8 @@ COPY --from=build_vllm ${COMMON_WORKDIR}/vllm /vllm-workspace
72
RUN cd /vllm-workspace \
73
&& rm -rf vllm \
74
&& python3 -m pip install -e tests/vllm_test_utils \
75
- && python3 -m pip install lm-eval[api]==0.4.4
+ && python3 -m pip install lm-eval[api]==0.4.4 \
76
+ && python3 -m pip install pytest-shard
77
78
# -----------------------
79
# Final vLLM image
0 commit comments