github
/
ChatGLM-6B
mirror of https://github.com/THUDM/ChatGLM-6B
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

git clone models

pull/666/head
bill.peng 2023-04-17 21:37:22 +08:00
parent d06b056050
commit efbc9e5e81
3 changed files with 20 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ptuning/docker/guide.sh
View File

@ -1,3 +1,10 @@
#!/usr/bin/env bash #!/usr/bin/env bash
# build docker for chatglm-6b training
docker build docker build
docker tag
docker login
docker push
docker run -it -

4
ptuning/docker/train_gpu/start.sh
View File

@ -4,7 +4,9 @@
source /app/dev/miniconda3/bin/activate base source /app/dev/miniconda3/bin/activate base
conda env list conda env list
# update source # update model and source
cd /app/source/ChatGLM-6B/ptuning/chatglm-6b || exit 1
git pull
cd /app/source/ChatGLM-6B/ptuning || exit 1 cd /app/source/ChatGLM-6B/ptuning || exit 1
git pull git pull

11
ptuning/docker/train_gpu/train.gpu.Dockerfile
View File

@ -15,7 +15,9 @@ ENV TZ=Asia/Shanghai
RUN apt-get -y update && apt-get install -y wget && \ RUN apt-get -y update && apt-get install -y wget && \
mkdir -p /app && wget -P /app https://repo.anaconda.com/miniconda/Miniconda3-py38_23.1.0-1-Linux-x86_64.sh && \ mkdir -p /app && wget -P /app https://repo.anaconda.com/miniconda/Miniconda3-py38_23.1.0-1-Linux-x86_64.sh && \
cd /app && bash Miniconda3-py38_23.1.0-1-Linux-x86_64.sh -b -p /app/dev/miniconda3 && \ cd /app && bash Miniconda3-py38_23.1.0-1-Linux-x86_64.sh -b -p /app/dev/miniconda3 && \
source /app/dev/miniconda3/bin/activate base && conda init bash && source ~/.bashrc && conda env list source /app/dev/miniconda3/bin/activate base && \
conda init bash && source ~/.bashrc && conda env list \
rm /app/Miniconda3-py38_23.1.0-1-Linux-x86_64.sh && ls -al /app/
# install chatglm-6b dependencies # install chatglm-6b dependencies
RUN source /app/dev/miniconda3/bin/activate base && conda env list && \ RUN source /app/dev/miniconda3/bin/activate base && conda env list && \
@ -25,7 +27,10 @@ RUN source /app/dev/miniconda3/bin/activate base && conda env list && \
gradio==3.20.0 mdtex2html fastapi uvicorn requests gradio==3.20.0 mdtex2html fastapi uvicorn requests
RUN mkdir -p /app/source && cd /app/source && \ RUN mkdir -p /app/source && cd /app/source && \
apt-get update -y && apt-get install -y git && \ apt-get update -y && apt-get install -y git && \
git clone --recursive https://github.com/zealotpb/ChatGLM-6B.git git clone --recursive https://github.com/zealotpb/ChatGLM-6B.git && \
cd /app/source/ChatGLM-6B/ptuning && \
apt-get update -y && apt-get install -y git-lfs && git lfs install && \
git clone --recursive https://huggingface.co/THUDM/chatglm-6b && ls -al chatglm-6b/
# copy start.sh # copy start.sh
COPY start.sh /usr/bin/start COPY start.sh /usr/bin/start
@ -33,3 +38,5 @@ RUN chmod a+x /usr/bin/start
# install necessary debug tools # install necessary debug tools
RUN apt-get update -y && apt-get install -y vim htop RUN apt-get update -y && apt-get install -y vim htop
CMD ["/bin/bash"]