Skip to content

Commit ff11ffb

Browse files
committed
Fix issue
Signed-off-by: ZePan110 <[email protected]>
1 parent 556b21b commit ff11ffb

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

comps/llms/src/text-generation/Dockerfile.intel_hpu_phi4

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,8 @@ COPY comps /home/user/comps
2121

2222
ARG uvpip='uv pip install --system --no-cache-dir'
2323
RUN pip install --no-cache-dir --upgrade pip setuptools uv && \
24-
pip install --no-cache-dir optimum-habana==1.18.0 && \
24+
pip install --no-cache-dir optimum-habana==1.18.1 && \
25+
pip install --no-cache-dir torch && \
2526
pip install --no-cache-dir git+https://github.com/HabanaAI/[email protected]
2627

2728
WORKDIR /home/user/comps/llms/src/text-generation

0 commit comments

Comments
 (0)