env

Author: jlamypoirier

Dockerfile

@@ -37,6 +37,8 @@ COPY --chown=$USERNAME transformers/ ./transformers
 # Stock version of pip doesn't work with editable transformers.
 RUN pip install --upgrade pip --no-cache-dir && pip install -r requirements.txt --no-cache-dir
 
+ENV HUGGINGFACE_HUB_CACHE=/app/data/.hf_cache/
+
 COPY --chown=$USERNAME Makefile .
 COPY --chown=$USERNAME src/ ./src
 COPY --chown=$USERNAME scripts/ ./scripts