Skip to content

Commit c2cb3e9

Browse files
xuechendiMu Huai
authored and
Mu Huai
committed
[CI]Fix hpu docker and numpy version for CI (vllm-project#16355)
Signed-off-by: Chendi Xue <chendi.xue@intel.com> Signed-off-by: Mu Huai <tianbowen.tbw@antgroup.com>
1 parent e410454 commit c2cb3e9

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

docker/Dockerfile.hpu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM vault.habana.ai/gaudi-docker/1.19.1/ubuntu22.04/habanalabs/pytorch-installer-2.5.1:latest
1+
FROM vault.habana.ai/gaudi-docker/1.20.1/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest
22

33
COPY ./ /workspace/vllm
44

requirements/hpu.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
ray
66
triton==3.1.0
77
pandas
8+
numpy==1.26.4
89
tabulate
910
setuptools>=61
1011
setuptools-scm>=8

0 commit comments

Comments
 (0)