Skip to content

Commit 421ddc4

Browse files
authored
Merge pull request #359 from vbedida79/patch-040225-1
tests_gaudi: Update vllm build for Gaudi 1.19.1
2 parents 72011e9 + c146594 commit 421ddc4

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tests/gaudi/l2/vllm_buildconfig.yaml

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,14 +20,17 @@ spec:
2020
runPolicy: "Serial"
2121
source:
2222
type: Dockerfile
23+
git:
24+
uri: https://github.com/HabanaAI/vllm-fork.git
25+
ref: v0.6.4.post2+Gaudi-1.19.0
2326
dockerfile: |
24-
ARG BASE_IMAGE=vault.habana.ai/gaudi-docker/1.18.0/rhel9.4/habanalabs/pytorch-installer-2.4.0:1.18.0-524
27+
ARG BASE_IMAGE=vault.habana.ai/gaudi-docker/1.19.1/rhel9.4/habanalabs/pytorch-installer-2.5.1:1.19.1-26
2528
FROM ${BASE_IMAGE} as habana-base
2629
2730
USER root
2831
2932
ENV VLLM_TARGET_DEVICE="hpu"
30-
ENV HABANA_SOFTWARE_VERSION="1.18.0-524"
33+
ENV HABANA_SOFTWARE_VERSION="1.19.1"
3134
3235
RUN dnf -y update --best --allowerasing --skip-broken && dnf clean all
3336
@@ -129,7 +132,7 @@ spec:
129132
dockerStrategy:
130133
buildArgs:
131134
- name: "BASE_IMAGE"
132-
value: "vault.habana.ai/gaudi-docker/1.18.0/rhel9.4/habanalabs/pytorch-installer-2.4.0:1.18.0-524"
135+
value: "vault.habana.ai/gaudi-docker/1.19.1/rhel9.4/habanalabs/pytorch-installer-2.5.1:1.19.1-26"
133136
output:
134137
to:
135138
kind: ImageStreamTag

0 commit comments

Comments
 (0)