changed structure

viraweb123 · Aug 14, 2024 · 4de85a6 · 4de85a6
1 parent 0d6701b
commit 4de85a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/Dockerfile → train/Dockerfile b/core/Dockerfile → train/Dockerfile
@@ -1,6 +1,6 @@
 FROM 139.99.239.192:5000/nvcr.io/nvidia/pytorch:23.12-py3
 
-ENV TRANSFORMERS_CACHE=/train/.cache/huggingface
+ENV HF_HOME=/train/.cache/huggingface
 
 RUN pip3 install --no-cache-dir transformers[torch]