Merge pull request axolotl-ai-cloud#283 from OpenAccess-AI-Collective…

…/docker-git-fetch git fetch fix for docker
mkeoliya · Jul 17, 2023 · db467a9 · db467a9
2 parents 0caa19f + 0da72e7
commit db467a9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git 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
diff --git 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 && \