tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

update entrypoint and force min accelerate

Browse Source
This commit is contained in:
Wing Lian
2023-05-18 06:25:34 -04:00
parent 13650732f8
commit fa8bd14be4
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
docker/Dockerfile-runpod
View File

@@ -1,11 +1,14 @@
ARG BASE_TAG=main ARG BASE_TAG=main
FROM winglian/axolotl:$BASE_TAG FROM winglian/axolotl:$BASE_TAG
COPY scripts/runpod-entrypoint.sh /root/runpod-entrypoint.sh
RUN apt install --yes --no-install-recommends openssh-server tmux && \ RUN apt install --yes --no-install-recommends openssh-server tmux && \
mkdir -p ~/.ssh && \ mkdir -p ~/.ssh && \
chmod 700 ~/.ssh && \ chmod 700 ~/.ssh && \
printf "\n[[ -z \"\$TMUX\" ]] && { tmux attach-session -t ssh_tmux || tmux new-session -s ssh_tmux; exit; }\n" >> ~/.bashrc && \ printf "\n[[ -z \"\$TMUX\" ]] && { tmux attach-session -t ssh_tmux || tmux new-session -s ssh_tmux; exit; }\n" >> ~/.bashrc && \
chmod +x /workspace/axolotl/scripts/runpod-entrypoint.sh chmod +x /workspace/axolotl/scripts/runpod-entrypoint.sh && \
chmod +x /root/runpod-entrypoint.sh
ENTRYPOINT ["/workspace/axolotl/scripts/runpod-entrypoint.sh"] ENTRYPOINT ["/root/runpod-entrypoint.sh"]
CMD ["sleep", "infinity"] CMD ["sleep", "infinity"]

2
requirements.txt
View File

@@ -6,7 +6,7 @@ PyYAML==6.0
black black
bitsandbytes==0.37.2 bitsandbytes==0.37.2
datasets datasets
accelerate accelerate>=0.19.0
sentencepiece sentencepiece
wandb wandb
einops einops