From eca3531329675cf5d5d0d0c2575ef6ee681691d6 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Sun, 16 Jul 2023 22:25:05 -0400
Subject: [PATCH 1/2] git fetch fix for docker

---
 docker/Dockerfile | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 38371c650..dcb6f9131 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -22,5 +22,9 @@ RUN cd axolotl && \
         pip install -e .; \
     fi
 
+# fix so that git fetch/pull from remote works
+RUN git config remote.origin.fetch "+refs/heads/*:refs/remotes/origin/*" && \
+    git config --get remote.origin.fetch
+
 # helper for huggingface-login cli
 RUN git config --global credential.helper store

From f162f3c7cca35fa97c3460e2d9f3b43dea900ebb Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Sun, 16 Jul 2023 23:03:54 -0400
Subject: [PATCH 2/2] set transformers cache env var in docker image

---
 docker/Dockerfile-runpod | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/docker/Dockerfile-runpod b/docker/Dockerfile-runpod
index a641f241e..2ea6e993c 100644
--- a/docker/Dockerfile-runpod
+++ b/docker/Dockerfile-runpod
@@ -1,6 +1,10 @@
 ARG BASE_TAG=main
 FROM winglian/axolotl:$BASE_TAG
 
+ENV HF_DATASETS_CACHE="/workspace/data/huggingface-cache/datasets"
+ENV HUGGINGFACE_HUB_CACHE="/workspace/data/huggingface-cache/hub"
+ENV TRANSFORMERS_CACHE="/workspace/data/huggingface-cache/hub"
+
 COPY scripts/runpod-entrypoint.sh /root/runpod-entrypoint.sh
 
 RUN apt install --yes --no-install-recommends openssh-server tmux && \