Skip to content

Commit d057ac8

Browse files
author
Le Yao
committed
update the Dockerfile to use latest fastchat
Signed-off-by: Le Yao <le.yao@intel.com>
1 parent b6b1b3a commit d057ac8

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

container/cnagc-fastchat/Dockerfile.compile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,4 +69,4 @@ ENV LD_PRELOAD "/opt/conda/lib/libiomp5.so /usr/lib/x86_64-linux-gnu/libtcmalloc
6969
ENV LD_LIBRARY_PATH "/opt/conda/lib/python3.8/site-packages/lib/"
7070

7171
ADD fastchat/ /fastchat
72-
RUN conda run --no-capture-output /opt/conda/bin/python -m pip install /fastchat/
72+
RUN conda run --no-capture-output /opt/conda/bin/python -m pip install -e /fastchat/".[model_worker,webui]"

container/cnagc-fastchat/Dockerfile.release

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,8 @@ ADD fastchat/ /fastchat
4343
FROM base as v2.0.100-cpu
4444

4545
RUN /opt/conda/bin/python -m pip install torch==2.0.1+cpu torchvision==0.15.2+cpu -f http://css-devops.sh.intel.com/download/mirror/torch/ --trusted-host css-devops.sh.intel.com
46-
RUN /opt/conda/bin/python -m pip install intel_extension_for_pytorch==2.0.100 prometheus_client
47-
RUN /opt/conda/bin/python -m pip install /fastchat/
46+
RUN /opt/conda/bin/python -m pip install intel_extension_for_pytorch==2.0.100 prometheus_client protobuf
47+
RUN /opt/conda/bin/python -m pip install -e /fastchat/".[model_worker,webui]"
4848

4949
# =======
5050

@@ -65,4 +65,4 @@ ENV LD_LIBRARY_PATH=/opt/intel/oneapi/mkl/2023.2.0/lib/intel64/:\
6565
RUN /opt/conda/bin/python -m pip install torch==2.0.1a0 torchvision==0.15.2a0 \
6666
intel_extension_for_pytorch==2.0.110+xpu \
6767
-f https://developer.intel.com/ipex-whl-stable-xpu
68-
RUN /opt/conda/bin/python -m pip install /fastchat/
68+
RUN /opt/conda/bin/python -m pip install -e /fastchat/".[model_worker,webui]"

0 commit comments

Comments
 (0)