set transformers cache env var in docker image
Browse files- docker/Dockerfile-runpod +4 -0
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 && \
|