update entrypoint and force min accelerate
Browse files- docker/Dockerfile-runpod +5 -2
- requirements.txt +1 -1
docker/Dockerfile-runpod
CHANGED
|
@@ -1,11 +1,14 @@
|
|
| 1 |
ARG BASE_TAG=main
|
| 2 |
FROM winglian/axolotl:$BASE_TAG
|
| 3 |
|
|
|
|
|
|
|
| 4 |
RUN apt install --yes --no-install-recommends openssh-server tmux && \
|
| 5 |
mkdir -p ~/.ssh && \
|
| 6 |
chmod 700 ~/.ssh && \
|
| 7 |
printf "\n[[ -z \"\$TMUX\" ]] && { tmux attach-session -t ssh_tmux || tmux new-session -s ssh_tmux; exit; }\n" >> ~/.bashrc && \
|
| 8 |
-
chmod +x /workspace/axolotl/scripts/runpod-entrypoint.sh
|
|
|
|
| 9 |
|
| 10 |
-
ENTRYPOINT ["/
|
| 11 |
CMD ["sleep", "infinity"]
|
|
|
|
| 1 |
ARG BASE_TAG=main
|
| 2 |
FROM winglian/axolotl:$BASE_TAG
|
| 3 |
|
| 4 |
+
COPY scripts/runpod-entrypoint.sh /root/runpod-entrypoint.sh
|
| 5 |
+
|
| 6 |
RUN apt install --yes --no-install-recommends openssh-server tmux && \
|
| 7 |
mkdir -p ~/.ssh && \
|
| 8 |
chmod 700 ~/.ssh && \
|
| 9 |
printf "\n[[ -z \"\$TMUX\" ]] && { tmux attach-session -t ssh_tmux || tmux new-session -s ssh_tmux; exit; }\n" >> ~/.bashrc && \
|
| 10 |
+
chmod +x /workspace/axolotl/scripts/runpod-entrypoint.sh && \
|
| 11 |
+
chmod +x /root/runpod-entrypoint.sh
|
| 12 |
|
| 13 |
+
ENTRYPOINT ["/root/runpod-entrypoint.sh"]
|
| 14 |
CMD ["sleep", "infinity"]
|
requirements.txt
CHANGED
|
@@ -6,7 +6,7 @@ PyYAML==6.0
|
|
| 6 |
black
|
| 7 |
bitsandbytes==0.37.2
|
| 8 |
datasets
|
| 9 |
-
accelerate
|
| 10 |
sentencepiece
|
| 11 |
wandb
|
| 12 |
einops
|
|
|
|
| 6 |
black
|
| 7 |
bitsandbytes==0.37.2
|
| 8 |
datasets
|
| 9 |
+
accelerate>=0.19.0
|
| 10 |
sentencepiece
|
| 11 |
wandb
|
| 12 |
einops
|