Update Dockerfile

runpod · Oct 24, 2023 · d25f69e · d25f69e
1 parent 2238ee6
commit d25f69e
Showing 1 changed file with 6 additions and 4 deletions.
diff --git 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 /