From 4df66f5cbc731666f504e826abb7c530a81bbf51 Mon Sep 17 00:00:00 2001 From: Ziteng Zhang <87107332+Jasonzzt@users.noreply.github.com> Date: Thu, 2 Nov 2023 16:26:24 +0800 Subject: [PATCH] Update llm-finetune-lora-cpu dockerfile and readme * Update README.md * Update Dockerfile --- docker/llm/finetune/lora/cpu/docker/Dockerfile | 4 ++-- docker/llm/finetune/lora/cpu/docker/README.md | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docker/llm/finetune/lora/cpu/docker/Dockerfile b/docker/llm/finetune/lora/cpu/docker/Dockerfile index eb08fd88..1d6d919d 100644 --- a/docker/llm/finetune/lora/cpu/docker/Dockerfile +++ b/docker/llm/finetune/lora/cpu/docker/Dockerfile @@ -24,7 +24,7 @@ RUN mkdir /bigdl/data && mkdir /bigdl/model && \ apt-get install -y python3-pip python3.9-dev python3-wheel git software-properties-common && \ pip3 install --upgrade pip && \ export PIP_DEFAULT_TIMEOUT=100 && \ - pip install --upgrade torch==2.0.1 --index-url https://download.pytorch.org/whl/cpu && \ + pip install --upgrade torch==2.1.0 --index-url https://download.pytorch.org/whl/cpu && \ # install ipex and oneccl pip install intel_extension_for_pytorch==2.0.100 && \ pip install oneccl_bind_pt -f https://developer.intel.com/ipex-whl-stable && \ @@ -61,4 +61,4 @@ COPY ./lora_finetune.py /bigdl/lora_finetune.py RUN chown -R mpiuser /bigdl USER mpiuser -ENTRYPOINT ["/bin/bash"] \ No newline at end of file +ENTRYPOINT ["/bin/bash"] diff --git a/docker/llm/finetune/lora/cpu/docker/README.md b/docker/llm/finetune/lora/cpu/docker/README.md index 2f3d5b76..4578dda6 100644 --- a/docker/llm/finetune/lora/cpu/docker/README.md +++ b/docker/llm/finetune/lora/cpu/docker/README.md @@ -33,7 +33,7 @@ docker run -itd \ -e WORKER_COUNT_DOCKER=your_worker_count \ -v your_downloaded_base_model_path:/bigdl/model \ -v your_downloaded_data_path:/bigdl/data/alpaca_data_cleaned_archive.json \ - intelanalytics/bigdl-llm-finetune-cpu:2.4.0-SNAPSHOT \ + intelanalytics/bigdl-llm-finetune-lora-cpu:2.4.0-SNAPSHOT \ bash ```