Skip to content

Commit f3bdbba

Browse files
committed
modify Dockerfile in finetuning
1 parent fb5e4a8 commit f3bdbba

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

neural_chat/docker/finetuning/Dockerfile

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
ARG UBUNTU_VER=22.04
3030
FROM ubuntu:${UBUNTU_VER} as cpu
3131

32-
ARG ITREX_VER=main
32+
ARG ITREX_VER=neural_chat_api
3333
ARG REPO=https://github.com/intel/intel-extension-for-transformers.git
3434

3535
# See http://bugs.python.org/issue19846
@@ -66,9 +66,9 @@ RUN source activate && conda activate chatbot-finetuning && pip install oneccl_b
6666
pip install datasets torch accelerate SentencePiece evaluate nltk rouge_score protobuf==3.20.1 tokenizers einops && \
6767
git clone https://github.com/huggingface/peft.git && cd peft && python setup.py install && \
6868
cd /itrex && pip install -v . && \
69-
cd workflows/chatbot/fine_tuning && pip install -r requirements.txt
69+
cd neural_chat/docker/finetuning/ && pip install -r requirements.txt
7070

71-
WORKDIR /itrex/neural_chat/examples/instruction_tuning/
71+
WORKDIR /itrex/neural_chat/docker/finetuning/
7272

7373

7474
# HABANA environment
@@ -92,15 +92,15 @@ RUN pip install git+https://github.com/huggingface/optimum-habana.git && \
9292
git clone https://github.com/huggingface/peft.git && cd peft && python setup.py install
9393

9494
# Download ITREX code
95-
ARG ITREX_VER=main
95+
ARG ITREX_VER=neural_chat_api
9696
ARG REPO=https://github.com/intel/intel-extension-for-transformers.git
9797

9898
RUN git clone --single-branch --branch=${ITREX_VER} ${REPO} itrex && \
99-
cd /itrex/workflows/chatbot/fine_tuning/ && \
99+
cd /itrex/neural_chat/docker/finetuning/ && \
100100
pip install -r requirements.txt
101101

102102
# Build ITREX
103103
RUN cd /itrex && pip install -v . && \
104104
pip install transformers==4.28.1
105105

106-
WORKDIR /itrex/neural_chat/examples/instruction_tuning
106+
WORKDIR /itrex/neural_chat/docker/finetuning/

0 commit comments

Comments
 (0)