cconsti commited on
Commit
bd03b19
·
verified ·
1 Parent(s): d81adb9

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -10,7 +10,8 @@ RUN pip install --no-cache-dir transformers datasets accelerate peft torch torch
10
  # Copy all local files into the container
11
  COPY . .
12
 
13
- ENV TRANSFORMERS_CACHE=/app/cache
14
- RUN mkdir -p /app/cache
 
15
  # Set default command (replace with your training script later)
16
  CMD ["python", "train.py"]
 
10
  # Copy all local files into the container
11
  COPY . .
12
 
13
+ # Set a writable cache directory inside the app folder
14
+ ENV HF_HOME=/app/hf_cache
15
+ RUN mkdir -p /app/hf_cache
16
  # Set default command (replace with your training script later)
17
  CMD ["python", "train.py"]