winglian commited on
Commit
0222466
2 Parent(s): 6f16c45 f162f3c

Merge pull request #283 from OpenAccess-AI-Collective/docker-git-fetch

Browse files
Files changed (2) hide show
  1. docker/Dockerfile +4 -0
  2. docker/Dockerfile-runpod +4 -0
docker/Dockerfile CHANGED
@@ -22,5 +22,9 @@ RUN cd axolotl && \
22
  pip install -e .; \
23
  fi
24
 
 
 
 
 
25
  # helper for huggingface-login cli
26
  RUN git config --global credential.helper store
 
22
  pip install -e .; \
23
  fi
24
 
25
+ # fix so that git fetch/pull from remote works
26
+ RUN git config remote.origin.fetch "+refs/heads/*:refs/remotes/origin/*" && \
27
+ git config --get remote.origin.fetch
28
+
29
  # helper for huggingface-login cli
30
  RUN git config --global credential.helper store
docker/Dockerfile-runpod CHANGED
@@ -1,6 +1,10 @@
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 && \
 
1
  ARG BASE_TAG=main
2
  FROM winglian/axolotl:$BASE_TAG
3
 
4
+ ENV HF_DATASETS_CACHE="/workspace/data/huggingface-cache/datasets"
5
+ ENV HUGGINGFACE_HUB_CACHE="/workspace/data/huggingface-cache/hub"
6
+ ENV TRANSFORMERS_CACHE="/workspace/data/huggingface-cache/hub"
7
+
8
  COPY scripts/runpod-entrypoint.sh /root/runpod-entrypoint.sh
9
 
10
  RUN apt install --yes --no-install-recommends openssh-server tmux && \