setup runpod images

use github.ref_name

docker/Dockerfile

@@ -4,7 +4,7 @@ FROM winglian/axolotl-base:$BASE_TAG
 ARG TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6+PTX"
 
 RUN apt-get update && \
-    apt-get install -y vim
+    apt-get install -y vim curl
 
 WORKDIR /workspace
 
@@ -16,3 +16,6 @@ RUN mkdir axolotl
 COPY . axolotl/
 RUN cd axolotl && \
     pip install -e .[int4]
+
+# helper for huggingface-login cli
+RUN git config --global credential.helper store

docker/Dockerfile-runpod

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