2929ARG UBUNTU_VER=22.04
3030FROM ubuntu:${UBUNTU_VER} as cpu
3131
32- ARG ITREX_VER=main
32+ ARG ITREX_VER=neural_chat_api
3333ARG 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
9696ARG REPO=https://github.com/intel/intel-extension-for-transformers.git
9797
9898RUN 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
103103RUN 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