Skip to content

Commit 2f84ae1

Browse files
authored
[CI/Build] Update LM Eval Version in AMD CI (vllm-project#27944)
Signed-off-by: zhewenli <zhewenli@meta.com>
1 parent f32cbc9 commit 2f84ae1

File tree

2 files changed

+9
-7
lines changed

2 files changed

+9
-7
lines changed

docker/Dockerfile.rocm

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,6 @@ COPY --from=build_vllm ${COMMON_WORKDIR}/vllm /vllm-workspace
7575
RUN cd /vllm-workspace \
7676
&& rm -rf vllm \
7777
&& python3 -m pip install -e tests/vllm_test_utils \
78-
&& python3 -m pip install lm-eval[api]==0.4.4 \
7978
&& python3 -m pip install pytest-shard
8079

8180
# -----------------------

requirements/rocm-test.txt

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ tblib==3.1.0
44
bm25s==0.2.13
55
pystemmer==3.0.0
66

7-
# entrypoints test
7+
# Entrypoints test
88
# librosa==0.10.2.post1 # required by audio tests in entrypoints/openai
99
audioread==3.0.1
1010
cffi==1.17.1
@@ -17,11 +17,11 @@ soundfile==0.13.1
1717
soxr==0.5.0.post1
1818
librosa==0.10.2.post1
1919

20-
# entrypoints test
20+
# Entrypoints test
2121
#vllm[video] # required by entrypoints/openai/test_video.py
2222
decord==0.6.0
2323

24-
# entrypoints test
24+
# Entrypoints test
2525
#sentence-transformers # required by entrypoints/openai/test_score.py
2626
sentence-transformers==3.4.1
2727

@@ -32,7 +32,10 @@ matplotlib==3.10.3
3232
blobfile==3.0.0
3333

3434
# Required for openai schema test.
35-
schemathesis==3.39.15
35+
schemathesis==3.39.15
3636

37-
# required for mteb test
38-
mteb[bm25s]>=1.38.11, <2
37+
# Required for mteb test
38+
mteb[bm25s]>=1.38.11, <2
39+
40+
# Required for eval tests
41+
lm-eval[api] @ git+https://github.com/EleutherAI/lm-evaluation-harness.git@206b7722158f58c35b7ffcd53b035fdbdda5126d

0 commit comments

Comments
 (0)