File tree Expand file tree Collapse file tree 1 file changed +3
-5
lines changed
.buildkite/scripts/hardware_ci Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change @@ -6,19 +6,17 @@ set -exuo pipefail
6
6
7
7
# Try building the docker image
8
8
cat << EOF | docker build -t hpu-plugin-v1-test-env -f - .
9
- FROM 1.22-413-pt2.7.1 :latest
9
+ FROM gaudi-base-image :latest
10
10
11
11
COPY ./ /workspace/vllm
12
12
13
13
WORKDIR /workspace/vllm
14
14
15
- RUN pip install -v -r requirements/hpu.txt
16
- RUN pip install git+https://github.com/vllm-project/vllm-gaudi.git
17
-
18
15
ENV no_proxy=localhost,127.0.0.1
19
16
ENV PT_HPU_ENABLE_LAZY_COLLECTIVES=true
20
17
21
- RUN VLLM_TARGET_DEVICE=hpu python3 setup.py install
18
+ RUN VLLM_TARGET_DEVICE=empty pip install .
19
+ RUN pip install git+https://github.com/vllm-project/vllm-gaudi.git
22
20
23
21
# install development dependencies (for testing)
24
22
RUN python3 -m pip install -e tests/vllm_test_utils
You can’t perform that action at this time.
0 commit comments