From fa8bd14be49ec234e28da1f444880bc9859d7589 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Thu, 18 May 2023 06:25:34 -0400
Subject: [PATCH] update entrypoint and force min accelerate

---
 docker/Dockerfile-runpod | 7 +++++--
 requirements.txt         | 2 +-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/docker/Dockerfile-runpod b/docker/Dockerfile-runpod
index f7dac3003..a641f241e 100644
--- a/docker/Dockerfile-runpod
+++ b/docker/Dockerfile-runpod
@@ -1,11 +1,14 @@
 ARG BASE_TAG=main
 FROM winglian/axolotl:$BASE_TAG
 
+COPY scripts/runpod-entrypoint.sh /root/runpod-entrypoint.sh
+
 RUN apt install --yes --no-install-recommends openssh-server tmux && \
     mkdir -p ~/.ssh && \
     chmod 700 ~/.ssh && \
     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"]
diff --git a/requirements.txt b/requirements.txt
index 13450f470..ae6689680 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,7 +6,7 @@ PyYAML==6.0
 black
 bitsandbytes==0.37.2
 datasets
-accelerate
+accelerate>=0.19.0
 sentencepiece
 wandb
 einops