Skip to content

Commit 5bcc41f

Browse files
authored
Upgrade HPU FW to 1.21; upgrade transformers to 4.51.3 (#608)
Signed-off-by: kaixuanliu <[email protected]>
1 parent be40719 commit 5bcc41f

File tree

4 files changed

+666
-667
lines changed

4 files changed

+666
-667
lines changed

Dockerfile-intel

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,9 +82,10 @@ RUN python -m pip install torch==2.7.0 torchvision torchaudio==2.7.0 --index-url
8282
RUN cd backends/python/server && \
8383
make install
8484

85-
FROM vault.habana.ai/gaudi-docker/1.19.0/ubuntu22.04/habanalabs/pytorch-installer-2.5.1:latest AS hpu
85+
FROM vault.habana.ai/gaudi-docker/1.21.0/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest AS hpu
8686
ENV HUGGINGFACE_HUB_CACHE=/data \
8787
PORT=80
88+
ENV PT_HPU_LAZY_MODE=1
8889

8990
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
9091
build-essential \

0 commit comments

Comments
 (0)