ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Update llm-finetune-lora-cpu dockerfile and readme

Browse source 
* Update README.md

* Update Dockerfile
This commit is contained in:
Ziteng Zhang 2023-11-02 16:26:24 +08:00 committed by GitHub
parent 2e3bfbfe1f
commit 4df66f5cbc
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
docker/llm/finetune/lora/cpu/docker/Dockerfile
View file

@ -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"]
ENTRYPOINT ["/bin/bash"]

2
docker/llm/finetune/lora/cpu/docker/README.md
View file

@ -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
```