From d25f69e85204ee1df03f9f4c9e26cfca21687fca Mon Sep 17 00:00:00 2001 From: Justin Merrell Date: Tue, 24 Oct 2023 19:43:07 -0400 Subject: [PATCH] Update Dockerfile --- official-templates/base/Dockerfile | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/official-templates/base/Dockerfile b/official-templates/base/Dockerfile index 0aa56f3..e20abe1 100644 --- a/official-templates/base/Dockerfile +++ b/official-templates/base/Dockerfile @@ -6,10 +6,12 @@ ENV SHELL=/bin/bash ENV PYTHONUNBUFFERED=True ENV DEBIAN_FRONTEND=noninteractive -ENV HF_HOME = "/runpod-volume/.cache/huggingface/" -ENV HF_DATASETS_CACHE = "/runpod-volume/.cache/huggingface/datasets/" -ENV DEFAULT_HF_METRICS_CACHE = "/runpod-volume/.cache/huggingface/metrics/" -ENV DEFAULT_HF_MODULES_CACHE = "/runpod-volume/.cache/huggingface/modules/" +ENV HF_HOME="/runpod-volume/.cache/huggingface/" +ENV HF_DATASETS_CACHE="/runpod-volume/.cache/huggingface/datasets/" +ENV DEFAULT_HF_METRICS_CACHE="/runpod-volume/.cache/huggingface/metrics/" +ENV DEFAULT_HF_MODULES_CACHE="/runpod-volume/.cache/huggingface/modules/" +ENV HUGGINGFACE_HUB_CACHE="/runpod-volume/.cache/huggingface/hub/" +ENV HUGGINGFACE_ASSETS_CACHE="/cache/huggingface/assets/" WORKDIR /