thechaiexperiment commited on
Commit
4deecd8
·
1 Parent(s): ffb1d75

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -1
Dockerfile CHANGED
@@ -26,7 +26,9 @@ RUN python -c "import nltk; nltk.download('punkt')"
26
  # Create and set permissions for the Transformers cache directory
27
  RUN mkdir -p /code/transformers_cache && chmod -R 777 /code/transformers_cache
28
  ENV TRANSFORMERS_CACHE=/code/transformers_cache
29
- ENV HF_HOME=/code/transformers_cache # Hugging Face cache
 
 
30
 
31
  # Download sentence-transformers model to avoid recreating it at runtime
32
  RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')"
 
26
  # Create and set permissions for the Transformers cache directory
27
  RUN mkdir -p /code/transformers_cache && chmod -R 777 /code/transformers_cache
28
  ENV TRANSFORMERS_CACHE=/code/transformers_cache
29
+
30
+ # Set HF_HOME for Hugging Face cache
31
+ ENV HF_HOME=/code/transformers_cache
32
 
33
  # Download sentence-transformers model to avoid recreating it at runtime
34
  RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')"