Skip to content
This repository has been archived by the owner on Oct 25, 2024. It is now read-only.

Bump neural docker base #1633

Open
wants to merge 4 commits into
base: main
Choose a base branch
from
Open

Bump neural docker base #1633

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
fbcb44b
Change base of habana image to 1.15.0 for neural chat docker
Chris-Sigopt Jun 21, 2024
08b9d5d
Update transformers to better version
Chris-Sigopt Jun 21, 2024
d1495c2
Bump to higher version
Chris-Sigopt Jun 21, 2024
551c145
Version of transformers that works with 1.15.0
Chris-Sigopt Jun 21, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions intel_extension_for_transformers/neural_chat/docker/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
# for more information.
#
# ============================================================================
# How to build:
# How to build:
# docker build ./ -f Dockerfile -t chatbot_finetune:latest
# If you need to use proxy, please use the following command
# docker build ./ --build-arg http_proxy=${http_proxy} --build-arg https_proxy=${http_proxy} -f Dockerfile -t chatbot_finetune:latest
Expand Down Expand Up @@ -82,7 +82,7 @@ WORKDIR /intel-extension-for-transformers/intel_extension_for_transformers/neura
CMD ["/usr/sbin/sshd", "-D"]

# HABANA environment
FROM vault.habana.ai/gaudi-docker/1.13.0/ubuntu22.04/habanalabs/pytorch-installer-2.1.0:latest as hpu
FROM vault.habana.ai/gaudi-docker/1.15.0/ubuntu22.04/habanalabs/pytorch-installer-2.2.0:latest as hpu

ENV LANG=en_US.UTF-8
ENV PYTHONPATH=/root:/usr/lib/habanalabs/
Expand Down Expand Up @@ -129,7 +129,7 @@ sed -i '/--extra-index-url https:\/\/download.pytorch.org\/whl\/cpu/d' requireme
pip install -r requirements.txt && \
cd /intel-extension-for-transformers/intel_extension_for_transformers/neural_chat/ && \
pip install -r requirements_hpu.txt && \
pip install transformers==4.34.1 && \
pip install transformers==4.38.2 && \
pip install accelerate==0.24.0 && \
pip install datasets==2.14.7

Expand Down
Loading